[2025-11-27 01:09:06] Namespace(method='FedProx', training_mode='head', dataset='mnist', 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/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 01:09:06] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:10:39] [Round 001] FedProx | Acc: 78.44% | Loss: 1.2239 | Time: 21.41s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:11:01] [Round 002] FedProx | Acc: 86.48% | Loss: 0.8565 | Time: 21.43s | Max GPU Mem: 1507.92 MB
[2025-11-27 01:11:23] [Round 003] FedProx | Acc: 88.38% | Loss: 0.6926 | Time: 22.29s | Max GPU Mem: 1506.79 MB
[2025-11-27 01:12:10] [Round 004] FedProx | Acc: 88.53% | Loss: 0.6009 | Time: 46.59s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:13:00] [Round 005] FedProx | Acc: 90.03% | Loss: 0.5342 | Time: 50.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:13:49] [Round 006] FedProx | Acc: 90.55% | Loss: 0.4917 | Time: 48.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:14:45] [Round 007] FedProx | Acc: 90.92% | Loss: 0.4570 | Time: 55.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:15:35] [Round 008] FedProx | Acc: 90.95% | Loss: 0.4307 | Time: 50.28s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:16:19] [Round 009] FedProx | Acc: 91.65% | Loss: 0.4097 | Time: 43.65s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:17:14] [Round 010] FedProx | Acc: 91.71% | Loss: 0.3897 | Time: 54.53s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:18:05] [Round 011] FedProx | Acc: 92.09% | Loss: 0.3758 | Time: 51.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:48] [Round 012] FedProx | Acc: 92.22% | Loss: 0.3609 | Time: 42.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:19:42] [Round 013] FedProx | Acc: 92.39% | Loss: 0.3507 | Time: 54.11s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:20:34] [Round 014] FedProx | Acc: 92.56% | Loss: 0.3376 | Time: 52.02s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:21:32] [Round 015] FedProx | Acc: 92.90% | Loss: 0.3274 | Time: 57.85s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:22:29] [Round 016] FedProx | Acc: 92.78% | Loss: 0.3205 | Time: 57.30s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:23:38] [Round 017] FedProx | Acc: 92.77% | Loss: 0.3127 | Time: 69.01s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:24:47] [Round 018] FedProx | Acc: 93.02% | Loss: 0.3040 | Time: 68.72s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:26:05] [Round 019] FedProx | Acc: 93.05% | Loss: 0.2978 | Time: 77.46s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:27:27] [Round 020] FedProx | Acc: 93.17% | Loss: 0.2915 | Time: 82.33s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:28:49] [Round 021] FedProx | Acc: 93.13% | Loss: 0.2871 | Time: 82.11s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:29:57] [Round 022] FedProx | Acc: 93.49% | Loss: 0.2801 | Time: 68.08s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:31:04] [Round 023] FedProx | Acc: 93.52% | Loss: 0.2756 | Time: 67.23s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:32:06] [Round 024] FedProx | Acc: 93.53% | Loss: 0.2726 | Time: 61.76s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:33:07] [Round 025] FedProx | Acc: 93.52% | Loss: 0.2675 | Time: 61.04s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:09] [Round 026] FedProx | Acc: 93.71% | Loss: 0.2627 | Time: 62.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:35:07] [Round 027] FedProx | Acc: 93.82% | Loss: 0.2587 | Time: 57.76s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:01] [Round 028] FedProx | Acc: 93.95% | Loss: 0.2550 | Time: 54.09s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:36:56] [Round 029] FedProx | Acc: 93.83% | Loss: 0.2527 | Time: 54.54s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:37:56] [Round 030] FedProx | Acc: 93.85% | Loss: 0.2492 | Time: 60.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:38:52] [Round 031] FedProx | Acc: 94.15% | Loss: 0.2446 | Time: 56.74s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:39:50] [Round 032] FedProx | Acc: 94.15% | Loss: 0.2428 | Time: 57.26s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:40:42] [Round 033] FedProx | Acc: 94.16% | Loss: 0.2400 | Time: 51.87s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:41:37] [Round 034] FedProx | Acc: 94.12% | Loss: 0.2386 | Time: 54.93s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:42:29] [Round 035] FedProx | Acc: 94.20% | Loss: 0.2357 | Time: 52.06s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:43:30] [Round 036] FedProx | Acc: 94.35% | Loss: 0.2322 | Time: 61.47s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:44:23] [Round 037] FedProx | Acc: 94.26% | Loss: 0.2297 | Time: 53.09s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:45:16] [Round 038] FedProx | Acc: 94.33% | Loss: 0.2280 | Time: 52.33s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:46:04] [Round 039] FedProx | Acc: 94.32% | Loss: 0.2258 | Time: 48.97s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:04] [Round 040] FedProx | Acc: 94.31% | Loss: 0.2241 | Time: 59.63s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:48:03] [Round 041] FedProx | Acc: 94.33% | Loss: 0.2245 | Time: 59.14s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:01] [Round 042] FedProx | Acc: 94.53% | Loss: 0.2198 | Time: 58.17s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:50:04] [Round 043] FedProx | Acc: 94.68% | Loss: 0.2172 | Time: 62.13s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:04] [Round 044] FedProx | Acc: 94.68% | Loss: 0.2163 | Time: 60.33s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:51:59] [Round 045] FedProx | Acc: 94.66% | Loss: 0.2149 | Time: 54.95s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:52:58] [Round 046] FedProx | Acc: 94.67% | Loss: 0.2119 | Time: 58.67s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:53:57] [Round 047] FedProx | Acc: 94.58% | Loss: 0.2115 | Time: 59.07s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:54:53] [Round 048] FedProx | Acc: 94.68% | Loss: 0.2100 | Time: 56.91s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:55:54] [Round 049] FedProx | Acc: 94.59% | Loss: 0.2103 | Time: 60.33s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:56:46] [Round 050] FedProx | Acc: 94.79% | Loss: 0.2062 | Time: 52.37s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:56:46] Saved to runs/run_20251127_001740_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s1.npz
