[2025-11-27 01:01:44] 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_001404_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.1)
[2025-11-27 01:01:44] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:03:47] [Round 001] FedProx | Acc: 37.40% | Loss: 2.1971 | Time: 9.97s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:03:58] [Round 002] FedProx | Acc: 43.93% | Loss: 1.5860 | Time: 11.19s | Max GPU Mem: 1507.27 MB
[2025-11-27 01:04:14] [Round 003] FedProx | Acc: 55.48% | Loss: 1.2603 | Time: 15.54s | Max GPU Mem: 1506.79 MB
[2025-11-27 01:04:28] [Round 004] FedProx | Acc: 57.33% | Loss: 1.3121 | Time: 13.91s | Max GPU Mem: 1507.31 MB
[2025-11-27 01:04:36] [Round 005] FedProx | Acc: 72.51% | Loss: 1.0034 | Time: 7.90s | Max GPU Mem: 1506.81 MB
[2025-11-27 01:04:48] [Round 006] FedProx | Acc: 75.17% | Loss: 0.8093 | Time: 12.66s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:05:53] [Round 007] FedProx | Acc: 75.23% | Loss: 0.8431 | Time: 64.77s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:06:41] [Round 008] FedProx | Acc: 75.50% | Loss: 0.7795 | Time: 48.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:07:21] [Round 009] FedProx | Acc: 75.42% | Loss: 0.7284 | Time: 39.25s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:08:16] [Round 010] FedProx | Acc: 78.68% | Loss: 0.6857 | Time: 55.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:09:09] [Round 011] FedProx | Acc: 80.55% | Loss: 0.6300 | Time: 52.82s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:09:46] [Round 012] FedProx | Acc: 78.13% | Loss: 0.6430 | Time: 36.44s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:10:50] [Round 013] FedProx | Acc: 81.00% | Loss: 0.6076 | Time: 64.37s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:11:18] [Round 014] FedProx | Acc: 73.08% | Loss: 0.9349 | Time: 28.16s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:12:11] [Round 015] FedProx | Acc: 82.65% | Loss: 0.5808 | Time: 53.04s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:12:42] [Round 016] FedProx | Acc: 65.68% | Loss: 0.8414 | Time: 30.38s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:13:50] [Round 017] FedProx | Acc: 79.12% | Loss: 0.6679 | Time: 68.66s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:14:56] [Round 018] FedProx | Acc: 83.10% | Loss: 0.5169 | Time: 65.55s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:15:51] [Round 019] FedProx | Acc: 67.59% | Loss: 0.7878 | Time: 55.08s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:16:56] [Round 020] FedProx | Acc: 82.35% | Loss: 0.5495 | Time: 64.73s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:18:25] [Round 021] FedProx | Acc: 79.21% | Loss: 0.5770 | Time: 89.27s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:19:28] [Round 022] FedProx | Acc: 84.93% | Loss: 0.5099 | Time: 63.30s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:20:40] [Round 023] FedProx | Acc: 81.28% | Loss: 0.5650 | Time: 71.91s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:21:41] [Round 024] FedProx | Acc: 85.40% | Loss: 0.4880 | Time: 60.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:22:38] [Round 025] FedProx | Acc: 82.77% | Loss: 0.5496 | Time: 56.78s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:23:45] [Round 026] FedProx | Acc: 81.25% | Loss: 0.5382 | Time: 67.33s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:24:33] [Round 027] FedProx | Acc: 87.90% | Loss: 0.4167 | Time: 47.91s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:25:25] [Round 028] FedProx | Acc: 83.26% | Loss: 0.4852 | Time: 52.56s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:26:03] [Round 029] FedProx | Acc: 78.85% | Loss: 0.5620 | Time: 37.19s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:27:27] [Round 030] FedProx | Acc: 83.64% | Loss: 0.4589 | Time: 84.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:28:35] [Round 031] FedProx | Acc: 83.73% | Loss: 0.5182 | Time: 67.28s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:29:29] [Round 032] FedProx | Acc: 82.75% | Loss: 0.4796 | Time: 54.51s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:30:39] [Round 033] FedProx | Acc: 89.63% | Loss: 0.3752 | Time: 70.06s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:31:39] [Round 034] FedProx | Acc: 78.81% | Loss: 0.5837 | Time: 59.43s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:32:35] [Round 035] FedProx | Acc: 81.93% | Loss: 0.5517 | Time: 56.10s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:33:45] [Round 036] FedProx | Acc: 82.57% | Loss: 0.5371 | Time: 70.20s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:34:38] [Round 037] FedProx | Acc: 86.37% | Loss: 0.4422 | Time: 52.81s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:35:36] [Round 038] FedProx | Acc: 87.68% | Loss: 0.3937 | Time: 58.50s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:36:16] [Round 039] FedProx | Acc: 83.70% | Loss: 0.4503 | Time: 39.65s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:37:18] [Round 040] FedProx | Acc: 89.87% | Loss: 0.3544 | Time: 62.35s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:38:14] [Round 041] FedProx | Acc: 89.00% | Loss: 0.3607 | Time: 55.35s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:39:37] [Round 042] FedProx | Acc: 84.98% | Loss: 0.4110 | Time: 83.13s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:40:15] [Round 043] FedProx | Acc: 89.41% | Loss: 0.3451 | Time: 37.99s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:41:06] [Round 044] FedProx | Acc: 89.34% | Loss: 0.3456 | Time: 51.30s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:42:28] [Round 045] FedProx | Acc: 88.40% | Loss: 0.3732 | Time: 81.55s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:43:30] [Round 046] FedProx | Acc: 90.67% | Loss: 0.3303 | Time: 62.24s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:44:10] [Round 047] FedProx | Acc: 76.38% | Loss: 0.5994 | Time: 39.80s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:44:57] [Round 048] FedProx | Acc: 88.28% | Loss: 0.3965 | Time: 47.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:45:34] [Round 049] FedProx | Acc: 90.12% | Loss: 0.3233 | Time: 36.30s | Max GPU Mem: 1506.83 MB
[2025-11-27 01:46:28] [Round 050] FedProx | Acc: 84.56% | Loss: 0.4655 | Time: 54.39s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:46:28] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s0.npz
