[2025-11-27 01:06:20] 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_001608_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 01:06:20] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:08:36] [Round 001] FedProx | Acc: 47.36% | Loss: 1.6344 | Time: 23.65s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:09:45] [Round 002] FedProx | Acc: 67.92% | Loss: 1.0661 | Time: 69.00s | Max GPU Mem: 1507.92 MB
[2025-11-27 01:10:43] [Round 003] FedProx | Acc: 80.64% | Loss: 0.9028 | Time: 58.01s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:12:02] [Round 004] FedProx | Acc: 73.97% | Loss: 0.8442 | Time: 78.90s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:13:00] [Round 005] FedProx | Acc: 81.94% | Loss: 0.6892 | Time: 58.16s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:14:34] [Round 006] FedProx | Acc: 85.09% | Loss: 0.6222 | Time: 93.70s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:15:46] [Round 007] FedProx | Acc: 79.30% | Loss: 0.6472 | Time: 72.42s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:16:53] [Round 008] FedProx | Acc: 86.40% | Loss: 0.5298 | Time: 67.21s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:17:46] [Round 009] FedProx | Acc: 89.29% | Loss: 0.4918 | Time: 52.80s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:52] [Round 010] FedProx | Acc: 87.62% | Loss: 0.4884 | Time: 65.70s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:20:14] [Round 011] FedProx | Acc: 88.68% | Loss: 0.4619 | Time: 82.60s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:21:17] [Round 012] FedProx | Acc: 89.65% | Loss: 0.4362 | Time: 62.26s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:22:02] [Round 013] FedProx | Acc: 89.45% | Loss: 0.4107 | Time: 45.61s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:23:01] [Round 014] FedProx | Acc: 85.80% | Loss: 0.4643 | Time: 58.94s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:24:08] [Round 015] FedProx | Acc: 88.29% | Loss: 0.4187 | Time: 66.40s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:25:04] [Round 016] FedProx | Acc: 91.38% | Loss: 0.3731 | Time: 56.65s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:26:13] [Round 017] FedProx | Acc: 91.27% | Loss: 0.3621 | Time: 68.52s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:27:16] [Round 018] FedProx | Acc: 88.68% | Loss: 0.3899 | Time: 63.32s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:28:10] [Round 019] FedProx | Acc: 91.97% | Loss: 0.3451 | Time: 53.97s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:28:55] [Round 020] FedProx | Acc: 90.08% | Loss: 0.3621 | Time: 44.61s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:29:48] [Round 021] FedProx | Acc: 91.87% | Loss: 0.3348 | Time: 53.31s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:30:45] [Round 022] FedProx | Acc: 91.77% | Loss: 0.3333 | Time: 57.17s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:31:35] [Round 023] FedProx | Acc: 91.99% | Loss: 0.3224 | Time: 49.73s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:32:25] [Round 024] FedProx | Acc: 91.48% | Loss: 0.3238 | Time: 50.41s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:33:40] [Round 025] FedProx | Acc: 90.48% | Loss: 0.3377 | Time: 74.57s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:41] [Round 026] FedProx | Acc: 92.77% | Loss: 0.3024 | Time: 61.63s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:35:32] [Round 027] FedProx | Acc: 93.15% | Loss: 0.2902 | Time: 50.86s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:28] [Round 028] FedProx | Acc: 91.31% | Loss: 0.3158 | Time: 55.63s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:37:14] [Round 029] FedProx | Acc: 92.59% | Loss: 0.2872 | Time: 46.17s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:38:07] [Round 030] FedProx | Acc: 91.07% | Loss: 0.3177 | Time: 52.45s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:39:37] [Round 031] FedProx | Acc: 93.11% | Loss: 0.2811 | Time: 90.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:44] [Round 032] FedProx | Acc: 92.99% | Loss: 0.2790 | Time: 66.89s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:42:01] [Round 033] FedProx | Acc: 92.92% | Loss: 0.2791 | Time: 76.81s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:43:25] [Round 034] FedProx | Acc: 93.18% | Loss: 0.2747 | Time: 84.32s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:44:48] [Round 035] FedProx | Acc: 92.79% | Loss: 0.2795 | Time: 83.28s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:46:09] [Round 036] FedProx | Acc: 93.57% | Loss: 0.2621 | Time: 80.69s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:47:09] [Round 037] FedProx | Acc: 90.73% | Loss: 0.3076 | Time: 60.43s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:48:06] [Round 038] FedProx | Acc: 92.49% | Loss: 0.2743 | Time: 56.45s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:49:06] [Round 039] FedProx | Acc: 92.67% | Loss: 0.2674 | Time: 60.31s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:53] [Round 040] FedProx | Acc: 92.98% | Loss: 0.2665 | Time: 47.01s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:50:46] [Round 041] FedProx | Acc: 92.88% | Loss: 0.2616 | Time: 53.28s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:40] [Round 042] FedProx | Acc: 93.29% | Loss: 0.2541 | Time: 54.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:52:39] [Round 043] FedProx | Acc: 93.41% | Loss: 0.2522 | Time: 59.06s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:53:38] [Round 044] FedProx | Acc: 91.84% | Loss: 0.2744 | Time: 58.53s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:54:44] [Round 045] FedProx | Acc: 93.58% | Loss: 0.2459 | Time: 65.94s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:55:15] [Round 046] FedProx | Acc: 92.65% | Loss: 0.2622 | Time: 30.70s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:55:34] [Round 047] FedProx | Acc: 93.45% | Loss: 0.2440 | Time: 19.27s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:55:54] [Round 048] FedProx | Acc: 92.86% | Loss: 0.2527 | Time: 19.91s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:56:15] [Round 049] FedProx | Acc: 93.47% | Loss: 0.2449 | Time: 21.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:56:26] [Round 050] FedProx | Acc: 93.86% | Loss: 0.2340 | Time: 10.52s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:56:26] Saved to runs/run_20251127_001608_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s0.npz
