[2025-11-27 01:06:49] 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_001608_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 01:06:49] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:09:29] [Round 001] FedProx | Acc: 52.05% | Loss: 1.7634 | Time: 57.64s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:10:32] [Round 002] FedProx | Acc: 55.67% | Loss: 1.3009 | Time: 62.75s | Max GPU Mem: 1507.92 MB
[2025-11-27 01:11:33] [Round 003] FedProx | Acc: 74.25% | Loss: 0.9984 | Time: 61.55s | Max GPU Mem: 1506.81 MB
[2025-11-27 01:12:20] [Round 004] FedProx | Acc: 73.76% | Loss: 0.9412 | Time: 46.25s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:13:33] [Round 005] FedProx | Acc: 84.88% | Loss: 0.7153 | Time: 73.04s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:15:03] [Round 006] FedProx | Acc: 88.66% | Loss: 0.5891 | Time: 90.66s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:16:11] [Round 007] FedProx | Acc: 87.72% | Loss: 0.5368 | Time: 67.43s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:17:27] [Round 008] FedProx | Acc: 87.53% | Loss: 0.5101 | Time: 76.02s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:18:33] [Round 009] FedProx | Acc: 89.53% | Loss: 0.4666 | Time: 65.88s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:19:31] [Round 010] FedProx | Acc: 89.78% | Loss: 0.4345 | Time: 58.26s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:20:38] [Round 011] FedProx | Acc: 89.97% | Loss: 0.4247 | Time: 66.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:21:37] [Round 012] FedProx | Acc: 91.51% | Loss: 0.3875 | Time: 58.91s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:22:43] [Round 013] FedProx | Acc: 91.45% | Loss: 0.3740 | Time: 66.80s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:23:53] [Round 014] FedProx | Acc: 91.61% | Loss: 0.3610 | Time: 69.31s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:24:50] [Round 015] FedProx | Acc: 92.16% | Loss: 0.3561 | Time: 57.39s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:25:49] [Round 016] FedProx | Acc: 91.57% | Loss: 0.3519 | Time: 59.11s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:26:52] [Round 017] FedProx | Acc: 92.30% | Loss: 0.3330 | Time: 62.88s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:27:53] [Round 018] FedProx | Acc: 92.68% | Loss: 0.3173 | Time: 60.73s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:28:43] [Round 019] FedProx | Acc: 92.33% | Loss: 0.3193 | Time: 49.76s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:29:35] [Round 020] FedProx | Acc: 93.33% | Loss: 0.2998 | Time: 52.05s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:30:20] [Round 021] FedProx | Acc: 92.02% | Loss: 0.3203 | Time: 45.06s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:31:04] [Round 022] FedProx | Acc: 91.51% | Loss: 0.3207 | Time: 44.02s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:31:51] [Round 023] FedProx | Acc: 93.48% | Loss: 0.2813 | Time: 47.19s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:32:46] [Round 024] FedProx | Acc: 92.86% | Loss: 0.2974 | Time: 54.85s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:33:41] [Round 025] FedProx | Acc: 92.77% | Loss: 0.3090 | Time: 54.92s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:30] [Round 026] FedProx | Acc: 92.80% | Loss: 0.2997 | Time: 49.41s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:35:20] [Round 027] FedProx | Acc: 92.73% | Loss: 0.2977 | Time: 50.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:14] [Round 028] FedProx | Acc: 93.44% | Loss: 0.2735 | Time: 53.92s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:37:02] [Round 029] FedProx | Acc: 93.09% | Loss: 0.2830 | Time: 47.40s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:37:54] [Round 030] FedProx | Acc: 93.16% | Loss: 0.2840 | Time: 52.32s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:38:52] [Round 031] FedProx | Acc: 93.43% | Loss: 0.2597 | Time: 58.10s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:07] [Round 032] FedProx | Acc: 93.58% | Loss: 0.2550 | Time: 74.59s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:41:12] [Round 033] FedProx | Acc: 93.73% | Loss: 0.2528 | Time: 65.60s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:41:59] [Round 034] FedProx | Acc: 93.40% | Loss: 0.2677 | Time: 46.78s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:42:54] [Round 035] FedProx | Acc: 93.43% | Loss: 0.2578 | Time: 55.23s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:43:46] [Round 036] FedProx | Acc: 93.88% | Loss: 0.2534 | Time: 52.00s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:44:34] [Round 037] FedProx | Acc: 93.36% | Loss: 0.2638 | Time: 47.99s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:45:30] [Round 038] FedProx | Acc: 93.95% | Loss: 0.2435 | Time: 55.99s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:46:11] [Round 039] FedProx | Acc: 93.72% | Loss: 0.2539 | Time: 40.62s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:45] [Round 040] FedProx | Acc: 93.68% | Loss: 0.2405 | Time: 94.40s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:48:45] [Round 041] FedProx | Acc: 93.07% | Loss: 0.2559 | Time: 59.89s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:36] [Round 042] FedProx | Acc: 93.83% | Loss: 0.2448 | Time: 50.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:50:40] [Round 043] FedProx | Acc: 93.43% | Loss: 0.2483 | Time: 63.83s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:40] [Round 044] FedProx | Acc: 94.43% | Loss: 0.2269 | Time: 59.60s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:52:41] [Round 045] FedProx | Acc: 93.93% | Loss: 0.2382 | Time: 61.45s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:53:44] [Round 046] FedProx | Acc: 93.50% | Loss: 0.2397 | Time: 62.82s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:55:02] [Round 047] FedProx | Acc: 94.26% | Loss: 0.2283 | Time: 77.80s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:55:19] [Round 048] FedProx | Acc: 93.82% | Loss: 0.2329 | Time: 16.91s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:55:45] [Round 049] FedProx | Acc: 94.28% | Loss: 0.2212 | Time: 26.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:56:11] [Round 050] FedProx | Acc: 93.97% | Loss: 0.2234 | Time: 26.00s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:56:11] Saved to runs/run_20251127_001608_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s1.npz
