[2025-11-27 05:10:04] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', model_name='resnet18', seed=0, rounds=50, num_clients=10, fraction=0.5, local_epochs=1, batch_size=64, lr=0.01, newton_lr=0.1, sophia_lr=0.05, rho=0.04, betas='0.9,0.99', mu=0.01, log_dir='runs/run_20251127_001608_resnet18_head/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 05:10:04] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:12:11] [Round 001] FedProx | Acc: 61.88% | Loss: 1.3409 | Time: 63.11s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:13:16] [Round 002] FedProx | Acc: 74.78% | Loss: 0.9588 | Time: 64.71s | Max GPU Mem: 1507.27 MB
[2025-11-27 05:14:34] [Round 003] FedProx | Acc: 77.26% | Loss: 0.7784 | Time: 78.09s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:16:08] [Round 004] FedProx | Acc: 78.66% | Loss: 0.6894 | Time: 94.08s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:16:30] [Round 005] FedProx | Acc: 80.19% | Loss: 0.6363 | Time: 21.35s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:16:52] [Round 006] FedProx | Acc: 78.37% | Loss: 0.6610 | Time: 22.73s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:17:13] [Round 007] FedProx | Acc: 78.45% | Loss: 0.6335 | Time: 21.03s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:17:27] [Round 008] FedProx | Acc: 79.75% | Loss: 0.5830 | Time: 13.33s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:17:38] [Round 009] FedProx | Acc: 81.26% | Loss: 0.5634 | Time: 11.40s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:17:49] [Round 010] FedProx | Acc: 79.86% | Loss: 0.5883 | Time: 10.82s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:18:00] [Round 011] FedProx | Acc: 79.55% | Loss: 0.5908 | Time: 10.74s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:18:15] [Round 012] FedProx | Acc: 81.58% | Loss: 0.5461 | Time: 15.71s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:18:33] [Round 013] FedProx | Acc: 83.18% | Loss: 0.5052 | Time: 17.10s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:18:57] [Round 014] FedProx | Acc: 83.54% | Loss: 0.4946 | Time: 24.41s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:19:57] [Round 015] FedProx | Acc: 82.25% | Loss: 0.5226 | Time: 60.00s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:20:57] [Round 016] FedProx | Acc: 83.80% | Loss: 0.4897 | Time: 60.26s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:22:10] [Round 017] FedProx | Acc: 82.65% | Loss: 0.5030 | Time: 72.52s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:23:10] [Round 018] FedProx | Acc: 81.78% | Loss: 0.5142 | Time: 60.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:24:10] [Round 019] FedProx | Acc: 83.26% | Loss: 0.4825 | Time: 60.60s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:25:00] [Round 020] FedProx | Acc: 82.16% | Loss: 0.5157 | Time: 49.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:25:57] [Round 021] FedProx | Acc: 83.90% | Loss: 0.4820 | Time: 56.53s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:26:53] [Round 022] FedProx | Acc: 83.22% | Loss: 0.4757 | Time: 55.88s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:27:56] [Round 023] FedProx | Acc: 82.84% | Loss: 0.4904 | Time: 62.98s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:28:57] [Round 024] FedProx | Acc: 84.16% | Loss: 0.4664 | Time: 61.15s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:29:49] [Round 025] FedProx | Acc: 84.12% | Loss: 0.4621 | Time: 52.60s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:30:22] [Round 026] FedProx | Acc: 83.28% | Loss: 0.4722 | Time: 32.59s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:31:22] [Round 027] FedProx | Acc: 84.32% | Loss: 0.4547 | Time: 60.42s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:32:29] [Round 028] FedProx | Acc: 84.26% | Loss: 0.4560 | Time: 66.27s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:33:12] [Round 029] FedProx | Acc: 84.79% | Loss: 0.4430 | Time: 43.80s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:34:14] [Round 030] FedProx | Acc: 84.55% | Loss: 0.4485 | Time: 61.82s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:35:19] [Round 031] FedProx | Acc: 84.30% | Loss: 0.4556 | Time: 65.18s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:36:12] [Round 032] FedProx | Acc: 84.92% | Loss: 0.4423 | Time: 52.82s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:37:24] [Round 033] FedProx | Acc: 83.48% | Loss: 0.4660 | Time: 71.87s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:38:09] [Round 034] FedProx | Acc: 84.78% | Loss: 0.4469 | Time: 44.91s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:39:12] [Round 035] FedProx | Acc: 85.01% | Loss: 0.4398 | Time: 62.69s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:40:38] [Round 036] FedProx | Acc: 83.89% | Loss: 0.4498 | Time: 86.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:42:12] [Round 037] FedProx | Acc: 84.82% | Loss: 0.4331 | Time: 94.69s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:43:42] [Round 038] FedProx | Acc: 85.09% | Loss: 0.4277 | Time: 89.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:44:56] [Round 039] FedProx | Acc: 84.18% | Loss: 0.4427 | Time: 74.09s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:45:55] [Round 040] FedProx | Acc: 84.19% | Loss: 0.4466 | Time: 58.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:46:45] [Round 041] FedProx | Acc: 84.76% | Loss: 0.4419 | Time: 49.94s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:47:46] [Round 042] FedProx | Acc: 84.97% | Loss: 0.4307 | Time: 60.97s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:48:40] [Round 043] FedProx | Acc: 83.45% | Loss: 0.4632 | Time: 53.89s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:49:29] [Round 044] FedProx | Acc: 84.70% | Loss: 0.4374 | Time: 49.37s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:50:33] [Round 045] FedProx | Acc: 84.90% | Loss: 0.4329 | Time: 63.99s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:51:38] [Round 046] FedProx | Acc: 85.17% | Loss: 0.4235 | Time: 64.56s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:52:23] [Round 047] FedProx | Acc: 85.64% | Loss: 0.4141 | Time: 45.48s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:53:11] [Round 048] FedProx | Acc: 85.67% | Loss: 0.4163 | Time: 48.05s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:54:14] [Round 049] FedProx | Acc: 85.96% | Loss: 0.4074 | Time: 62.66s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:54:58] [Round 050] FedProx | Acc: 84.59% | Loss: 0.4373 | Time: 44.17s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:54:58] Saved to runs/run_20251127_001608_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s0.npz
