[2025-11-27 01:08:47] Namespace(method='FedProx', training_mode='head', dataset='mnist', 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_001740_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 01:08:47] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:10:01] [Round 001] FedProx | Acc: 79.95% | Loss: 1.2035 | Time: 11.99s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:10:12] [Round 002] FedProx | Acc: 85.74% | Loss: 0.8691 | Time: 11.16s | Max GPU Mem: 1507.92 MB
[2025-11-27 01:10:29] [Round 003] FedProx | Acc: 87.81% | Loss: 0.7042 | Time: 17.06s | Max GPU Mem: 1506.79 MB
[2025-11-27 01:10:51] [Round 004] FedProx | Acc: 88.85% | Loss: 0.6081 | Time: 21.55s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:11:13] [Round 005] FedProx | Acc: 89.69% | Loss: 0.5424 | Time: 22.41s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:11:44] [Round 006] FedProx | Acc: 90.60% | Loss: 0.4930 | Time: 30.57s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:12:55] [Round 007] FedProx | Acc: 91.14% | Loss: 0.4582 | Time: 71.24s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:13:56] [Round 008] FedProx | Acc: 91.22% | Loss: 0.4343 | Time: 61.16s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:15:09] [Round 009] FedProx | Acc: 91.64% | Loss: 0.4110 | Time: 72.40s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:16:21] [Round 010] FedProx | Acc: 91.90% | Loss: 0.3919 | Time: 71.82s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:17:40] [Round 011] FedProx | Acc: 92.10% | Loss: 0.3765 | Time: 79.33s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:46] [Round 012] FedProx | Acc: 91.92% | Loss: 0.3642 | Time: 66.32s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:19:50] [Round 013] FedProx | Acc: 92.61% | Loss: 0.3500 | Time: 63.83s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:20:54] [Round 014] FedProx | Acc: 92.60% | Loss: 0.3392 | Time: 63.70s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:21:58] [Round 015] FedProx | Acc: 92.81% | Loss: 0.3313 | Time: 63.95s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:22:58] [Round 016] FedProx | Acc: 92.78% | Loss: 0.3201 | Time: 60.67s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:23:56] [Round 017] FedProx | Acc: 93.09% | Loss: 0.3118 | Time: 57.53s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:24:52] [Round 018] FedProx | Acc: 93.22% | Loss: 0.3046 | Time: 55.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:25:55] [Round 019] FedProx | Acc: 93.27% | Loss: 0.2984 | Time: 63.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:26:53] [Round 020] FedProx | Acc: 93.16% | Loss: 0.2925 | Time: 57.55s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:27:50] [Round 021] FedProx | Acc: 93.23% | Loss: 0.2867 | Time: 57.91s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:28:51] [Round 022] FedProx | Acc: 93.51% | Loss: 0.2813 | Time: 60.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:29:55] [Round 023] FedProx | Acc: 93.42% | Loss: 0.2761 | Time: 63.57s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:30:55] [Round 024] FedProx | Acc: 93.72% | Loss: 0.2714 | Time: 59.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:31:55] [Round 025] FedProx | Acc: 93.51% | Loss: 0.2683 | Time: 60.84s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:32:57] [Round 026] FedProx | Acc: 93.71% | Loss: 0.2627 | Time: 61.58s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:33:54] [Round 027] FedProx | Acc: 93.55% | Loss: 0.2605 | Time: 57.27s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:52] [Round 028] FedProx | Acc: 93.73% | Loss: 0.2561 | Time: 57.25s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:35:53] [Round 029] FedProx | Acc: 93.94% | Loss: 0.2523 | Time: 61.75s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:53] [Round 030] FedProx | Acc: 93.70% | Loss: 0.2513 | Time: 59.25s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:37:58] [Round 031] FedProx | Acc: 94.04% | Loss: 0.2468 | Time: 65.76s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:39:04] [Round 032] FedProx | Acc: 93.96% | Loss: 0.2440 | Time: 65.44s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:40:07] [Round 033] FedProx | Acc: 94.06% | Loss: 0.2405 | Time: 63.49s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:41:12] [Round 034] FedProx | Acc: 94.29% | Loss: 0.2369 | Time: 64.59s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:42:04] [Round 035] FedProx | Acc: 93.98% | Loss: 0.2374 | Time: 52.49s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:43:02] [Round 036] FedProx | Acc: 94.06% | Loss: 0.2328 | Time: 57.54s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:43:57] [Round 037] FedProx | Acc: 94.38% | Loss: 0.2310 | Time: 54.77s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:44:49] [Round 038] FedProx | Acc: 94.42% | Loss: 0.2277 | Time: 52.09s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:45:50] [Round 039] FedProx | Acc: 94.56% | Loss: 0.2255 | Time: 61.50s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:46:40] [Round 040] FedProx | Acc: 94.36% | Loss: 0.2238 | Time: 49.99s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:47:39] [Round 041] FedProx | Acc: 94.47% | Loss: 0.2211 | Time: 58.53s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:48:56] [Round 042] FedProx | Acc: 94.54% | Loss: 0.2198 | Time: 77.58s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:50:06] [Round 043] FedProx | Acc: 94.55% | Loss: 0.2186 | Time: 69.33s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:17] [Round 044] FedProx | Acc: 94.57% | Loss: 0.2171 | Time: 71.01s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:52:37] [Round 045] FedProx | Acc: 94.61% | Loss: 0.2151 | Time: 80.35s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:53:55] [Round 046] FedProx | Acc: 94.78% | Loss: 0.2129 | Time: 77.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:54:57] [Round 047] FedProx | Acc: 94.83% | Loss: 0.2108 | Time: 62.12s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:56:01] [Round 048] FedProx | Acc: 94.87% | Loss: 0.2087 | Time: 64.28s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:56:54] [Round 049] FedProx | Acc: 94.68% | Loss: 0.2081 | Time: 53.02s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:57:16] [Round 050] FedProx | Acc: 94.72% | Loss: 0.2072 | Time: 21.54s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:57:16] Saved to runs/run_20251127_001740_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s0.npz
