[2025-11-27 05:26:08] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', model_name='resnet18', seed=1, 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_001740_resnet18_head/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 05:26:08] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:28:36] [Round 001] FedProx | Acc: 78.83% | Loss: 0.9326 | Time: 36.88s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:29:33] [Round 002] FedProx | Acc: 79.77% | Loss: 0.7390 | Time: 56.89s | Max GPU Mem: 1507.31 MB
[2025-11-27 05:30:45] [Round 003] FedProx | Acc: 80.61% | Loss: 0.6550 | Time: 71.50s | Max GPU Mem: 1506.81 MB
[2025-11-27 05:32:00] [Round 004] FedProx | Acc: 81.35% | Loss: 0.6088 | Time: 75.79s | Max GPU Mem: 1507.31 MB
[2025-11-27 05:33:14] [Round 005] FedProx | Acc: 82.07% | Loss: 0.5776 | Time: 73.81s | Max GPU Mem: 1506.81 MB
[2025-11-27 05:34:23] [Round 006] FedProx | Acc: 82.06% | Loss: 0.5597 | Time: 68.83s | Max GPU Mem: 1507.31 MB
[2025-11-27 05:35:40] [Round 007] FedProx | Acc: 82.73% | Loss: 0.5393 | Time: 76.99s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:36:51] [Round 008] FedProx | Acc: 83.08% | Loss: 0.5240 | Time: 70.87s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:38:01] [Round 009] FedProx | Acc: 83.28% | Loss: 0.5122 | Time: 70.00s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:39:07] [Round 010] FedProx | Acc: 83.58% | Loss: 0.5024 | Time: 66.53s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:40:13] [Round 011] FedProx | Acc: 83.56% | Loss: 0.4945 | Time: 65.39s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:41:20] [Round 012] FedProx | Acc: 83.88% | Loss: 0.4858 | Time: 67.39s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:42:26] [Round 013] FedProx | Acc: 84.18% | Loss: 0.4798 | Time: 65.30s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:43:28] [Round 014] FedProx | Acc: 84.12% | Loss: 0.4743 | Time: 62.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:44:30] [Round 015] FedProx | Acc: 84.34% | Loss: 0.4670 | Time: 61.30s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:45:28] [Round 016] FedProx | Acc: 84.53% | Loss: 0.4609 | Time: 58.42s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:46:25] [Round 017] FedProx | Acc: 84.65% | Loss: 0.4564 | Time: 56.67s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:47:17] [Round 018] FedProx | Acc: 84.61% | Loss: 0.4531 | Time: 52.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:48:08] [Round 019] FedProx | Acc: 84.77% | Loss: 0.4508 | Time: 50.93s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:48:59] [Round 020] FedProx | Acc: 85.02% | Loss: 0.4452 | Time: 51.04s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:49:51] [Round 021] FedProx | Acc: 85.12% | Loss: 0.4437 | Time: 52.23s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:50:46] [Round 022] FedProx | Acc: 85.24% | Loss: 0.4390 | Time: 54.58s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:51:43] [Round 023] FedProx | Acc: 85.29% | Loss: 0.4363 | Time: 57.22s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:52:39] [Round 024] FedProx | Acc: 85.37% | Loss: 0.4342 | Time: 56.14s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:53:35] [Round 025] FedProx | Acc: 85.34% | Loss: 0.4317 | Time: 55.56s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:54:27] [Round 026] FedProx | Acc: 85.46% | Loss: 0.4282 | Time: 52.59s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:55:21] [Round 027] FedProx | Acc: 85.29% | Loss: 0.4264 | Time: 53.48s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:56:06] [Round 028] FedProx | Acc: 85.45% | Loss: 0.4233 | Time: 45.69s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:57:04] [Round 029] FedProx | Acc: 85.51% | Loss: 0.4210 | Time: 57.44s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:57:55] [Round 030] FedProx | Acc: 85.74% | Loss: 0.4191 | Time: 51.38s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:58:48] [Round 031] FedProx | Acc: 85.61% | Loss: 0.4178 | Time: 53.21s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:59:49] [Round 032] FedProx | Acc: 85.84% | Loss: 0.4153 | Time: 60.78s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:00:37] [Round 033] FedProx | Acc: 85.75% | Loss: 0.4139 | Time: 47.62s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:01:28] [Round 034] FedProx | Acc: 85.85% | Loss: 0.4120 | Time: 51.69s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:02:26] [Round 035] FedProx | Acc: 85.97% | Loss: 0.4101 | Time: 57.34s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:03:37] [Round 036] FedProx | Acc: 86.02% | Loss: 0.4080 | Time: 71.04s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:04:44] [Round 037] FedProx | Acc: 85.99% | Loss: 0.4070 | Time: 67.46s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:05:36] [Round 038] FedProx | Acc: 86.13% | Loss: 0.4051 | Time: 51.64s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:06:19] [Round 039] FedProx | Acc: 86.14% | Loss: 0.4047 | Time: 42.84s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:07:09] [Round 040] FedProx | Acc: 86.12% | Loss: 0.4019 | Time: 50.17s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:08:09] [Round 041] FedProx | Acc: 86.25% | Loss: 0.4012 | Time: 59.84s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:09:02] [Round 042] FedProx | Acc: 86.19% | Loss: 0.3998 | Time: 53.25s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:09:28] [Round 043] FedProx | Acc: 86.41% | Loss: 0.3987 | Time: 26.32s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:09:50] [Round 044] FedProx | Acc: 86.20% | Loss: 0.3994 | Time: 21.36s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:10:10] [Round 045] FedProx | Acc: 86.44% | Loss: 0.3968 | Time: 20.65s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:10:32] [Round 046] FedProx | Acc: 86.52% | Loss: 0.3962 | Time: 21.45s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:10:50] [Round 047] FedProx | Acc: 86.50% | Loss: 0.3939 | Time: 18.32s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:11:10] [Round 048] FedProx | Acc: 86.47% | Loss: 0.3931 | Time: 19.79s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:11:56] [Round 049] FedProx | Acc: 86.39% | Loss: 0.3941 | Time: 46.54s | Max GPU Mem: 1506.83 MB
[2025-11-27 06:12:49] [Round 050] FedProx | Acc: 86.62% | Loss: 0.3914 | Time: 52.07s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:12:49] Saved to runs/run_20251127_001740_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s1.npz
