[2025-11-27 01:03:01] Namespace(method='FedProx', training_mode='head', dataset='mnist', model_name='resnet18', seed=2, 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:03:01] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:05:57] [Round 001] FedProx | Acc: 29.02% | Loss: 2.2319 | Time: 64.83s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:07:19] [Round 002] FedProx | Acc: 28.39% | Loss: 2.0153 | Time: 81.56s | Max GPU Mem: 1507.90 MB
[2025-11-27 01:08:23] [Round 003] FedProx | Acc: 35.22% | Loss: 1.7300 | Time: 64.07s | Max GPU Mem: 1506.77 MB
[2025-11-27 01:09:46] [Round 004] FedProx | Acc: 54.47% | Loss: 1.2076 | Time: 83.64s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:11:02] [Round 005] FedProx | Acc: 51.24% | Loss: 1.4311 | Time: 75.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:12:38] [Round 006] FedProx | Acc: 55.43% | Loss: 1.2290 | Time: 96.12s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:13:51] [Round 007] FedProx | Acc: 70.30% | Loss: 0.9770 | Time: 73.09s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:15:16] [Round 008] FedProx | Acc: 59.70% | Loss: 1.1210 | Time: 84.81s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:16:21] [Round 009] FedProx | Acc: 74.44% | Loss: 0.8150 | Time: 65.47s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:17:39] [Round 010] FedProx | Acc: 79.06% | Loss: 0.7353 | Time: 77.40s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:18:40] [Round 011] FedProx | Acc: 75.89% | Loss: 0.7626 | Time: 61.50s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:19:43] [Round 012] FedProx | Acc: 76.16% | Loss: 0.7255 | Time: 62.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:20:44] [Round 013] FedProx | Acc: 82.97% | Loss: 0.6043 | Time: 60.60s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:21:40] [Round 014] FedProx | Acc: 78.30% | Loss: 0.7094 | Time: 56.53s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:22:35] [Round 015] FedProx | Acc: 77.20% | Loss: 0.7301 | Time: 54.33s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:23:29] [Round 016] FedProx | Acc: 75.73% | Loss: 0.7269 | Time: 54.69s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:24:15] [Round 017] FedProx | Acc: 83.95% | Loss: 0.5415 | Time: 45.41s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:25:17] [Round 018] FedProx | Acc: 82.19% | Loss: 0.5750 | Time: 62.29s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:25:58] [Round 019] FedProx | Acc: 85.20% | Loss: 0.5119 | Time: 41.03s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:26:45] [Round 020] FedProx | Acc: 80.84% | Loss: 0.5841 | Time: 46.91s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:27:41] [Round 021] FedProx | Acc: 80.10% | Loss: 0.6523 | Time: 55.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:28:42] [Round 022] FedProx | Acc: 84.36% | Loss: 0.5164 | Time: 61.67s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:29:40] [Round 023] FedProx | Acc: 89.25% | Loss: 0.4356 | Time: 58.10s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:30:38] [Round 024] FedProx | Acc: 83.29% | Loss: 0.5158 | Time: 57.84s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:31:12] [Round 025] FedProx | Acc: 83.46% | Loss: 0.4906 | Time: 33.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:31:55] [Round 026] FedProx | Acc: 86.87% | Loss: 0.4626 | Time: 43.12s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:32:44] [Round 027] FedProx | Acc: 90.40% | Loss: 0.3965 | Time: 48.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:33:40] [Round 028] FedProx | Acc: 84.70% | Loss: 0.4942 | Time: 56.55s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:34:42] [Round 029] FedProx | Acc: 85.84% | Loss: 0.4806 | Time: 61.25s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:35:35] [Round 030] FedProx | Acc: 84.75% | Loss: 0.5059 | Time: 53.71s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:36:19] [Round 031] FedProx | Acc: 80.39% | Loss: 0.5577 | Time: 43.25s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:37:04] [Round 032] FedProx | Acc: 84.59% | Loss: 0.4588 | Time: 45.11s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:38:14] [Round 033] FedProx | Acc: 86.33% | Loss: 0.4516 | Time: 69.94s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:39:23] [Round 034] FedProx | Acc: 85.33% | Loss: 0.4639 | Time: 68.95s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:39:57] [Round 035] FedProx | Acc: 85.18% | Loss: 0.4598 | Time: 34.74s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:53] [Round 036] FedProx | Acc: 85.47% | Loss: 0.4700 | Time: 56.16s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:41:46] [Round 037] FedProx | Acc: 88.56% | Loss: 0.3946 | Time: 52.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:42:28] [Round 038] FedProx | Acc: 86.78% | Loss: 0.4101 | Time: 42.23s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:43:30] [Round 039] FedProx | Acc: 86.53% | Loss: 0.4265 | Time: 61.83s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:44:25] [Round 040] FedProx | Acc: 86.96% | Loss: 0.4468 | Time: 54.75s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:45:16] [Round 041] FedProx | Acc: 87.26% | Loss: 0.4073 | Time: 51.13s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:46:05] [Round 042] FedProx | Acc: 86.87% | Loss: 0.4003 | Time: 49.46s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:46:43] [Round 043] FedProx | Acc: 85.52% | Loss: 0.4411 | Time: 38.19s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:04] [Round 044] FedProx | Acc: 81.91% | Loss: 0.5163 | Time: 20.39s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:47:19] [Round 045] FedProx | Acc: 87.80% | Loss: 0.3838 | Time: 15.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:43] [Round 046] FedProx | Acc: 83.30% | Loss: 0.4896 | Time: 23.89s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:48:04] [Round 047] FedProx | Acc: 84.62% | Loss: 0.4686 | Time: 21.01s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:48:21] [Round 048] FedProx | Acc: 88.79% | Loss: 0.3744 | Time: 17.00s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:49:13] [Round 049] FedProx | Acc: 86.32% | Loss: 0.4262 | Time: 51.84s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:50:09] [Round 050] FedProx | Acc: 86.29% | Loss: 0.4007 | Time: 55.67s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:50:09] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s2.npz
