[2025-11-27 01:09:38] 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_001740_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 01:09:38] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:12:35] [Round 001] FedProx | Acc: 81.36% | Loss: 1.2215 | Time: 64.79s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:13:46] [Round 002] FedProx | Acc: 85.95% | Loss: 0.8653 | Time: 70.63s | Max GPU Mem: 1507.94 MB
[2025-11-27 01:14:48] [Round 003] FedProx | Acc: 87.92% | Loss: 0.7111 | Time: 62.30s | Max GPU Mem: 1506.81 MB
[2025-11-27 01:16:01] [Round 004] FedProx | Acc: 89.18% | Loss: 0.6101 | Time: 73.20s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:17:14] [Round 005] FedProx | Acc: 90.09% | Loss: 0.5433 | Time: 73.08s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:22] [Round 006] FedProx | Acc: 90.21% | Loss: 0.5002 | Time: 67.40s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:19:36] [Round 007] FedProx | Acc: 90.91% | Loss: 0.4667 | Time: 74.08s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:20:37] [Round 008] FedProx | Acc: 91.01% | Loss: 0.4396 | Time: 61.02s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:21:44] [Round 009] FedProx | Acc: 91.16% | Loss: 0.4176 | Time: 67.70s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:22:45] [Round 010] FedProx | Acc: 91.73% | Loss: 0.3944 | Time: 60.51s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:23:41] [Round 011] FedProx | Acc: 92.11% | Loss: 0.3782 | Time: 56.60s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:24:47] [Round 012] FedProx | Acc: 92.04% | Loss: 0.3660 | Time: 65.36s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:25:53] [Round 013] FedProx | Acc: 92.55% | Loss: 0.3514 | Time: 65.83s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:26:52] [Round 014] FedProx | Acc: 92.42% | Loss: 0.3430 | Time: 59.15s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:27:50] [Round 015] FedProx | Acc: 92.68% | Loss: 0.3320 | Time: 58.21s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:28:51] [Round 016] FedProx | Acc: 92.72% | Loss: 0.3258 | Time: 61.00s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:29:56] [Round 017] FedProx | Acc: 92.92% | Loss: 0.3174 | Time: 65.21s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:31:02] [Round 018] FedProx | Acc: 92.84% | Loss: 0.3090 | Time: 66.22s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:32:04] [Round 019] FedProx | Acc: 92.99% | Loss: 0.3025 | Time: 61.89s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:33:00] [Round 020] FedProx | Acc: 93.13% | Loss: 0.2945 | Time: 55.61s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:33:57] [Round 021] FedProx | Acc: 93.41% | Loss: 0.2885 | Time: 57.28s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:55] [Round 022] FedProx | Acc: 93.51% | Loss: 0.2839 | Time: 58.11s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:35:54] [Round 023] FedProx | Acc: 93.55% | Loss: 0.2776 | Time: 59.17s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:53] [Round 024] FedProx | Acc: 93.55% | Loss: 0.2737 | Time: 58.12s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:37:54] [Round 025] FedProx | Acc: 93.44% | Loss: 0.2703 | Time: 61.07s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:38:48] [Round 026] FedProx | Acc: 93.52% | Loss: 0.2663 | Time: 54.00s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:39:42] [Round 027] FedProx | Acc: 93.77% | Loss: 0.2607 | Time: 54.14s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:32] [Round 028] FedProx | Acc: 93.74% | Loss: 0.2580 | Time: 50.48s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:41:23] [Round 029] FedProx | Acc: 93.90% | Loss: 0.2547 | Time: 50.19s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:42:25] [Round 030] FedProx | Acc: 93.83% | Loss: 0.2513 | Time: 62.86s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:43:32] [Round 031] FedProx | Acc: 93.97% | Loss: 0.2476 | Time: 66.96s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:44:47] [Round 032] FedProx | Acc: 93.94% | Loss: 0.2455 | Time: 74.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:46:00] [Round 033] FedProx | Acc: 94.03% | Loss: 0.2414 | Time: 73.34s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:05] [Round 034] FedProx | Acc: 94.10% | Loss: 0.2402 | Time: 64.41s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:48:04] [Round 035] FedProx | Acc: 94.10% | Loss: 0.2370 | Time: 59.03s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:02] [Round 036] FedProx | Acc: 94.24% | Loss: 0.2345 | Time: 57.88s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:50:04] [Round 037] FedProx | Acc: 94.16% | Loss: 0.2325 | Time: 61.96s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:04] [Round 038] FedProx | Acc: 94.31% | Loss: 0.2294 | Time: 60.15s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:51:59] [Round 039] FedProx | Acc: 94.24% | Loss: 0.2275 | Time: 55.39s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:52:58] [Round 040] FedProx | Acc: 94.39% | Loss: 0.2245 | Time: 59.09s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:54:00] [Round 041] FedProx | Acc: 94.53% | Loss: 0.2232 | Time: 61.65s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:54:58] [Round 042] FedProx | Acc: 94.38% | Loss: 0.2224 | Time: 58.16s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:55:59] [Round 043] FedProx | Acc: 94.32% | Loss: 0.2207 | Time: 61.11s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:56:51] [Round 044] FedProx | Acc: 94.64% | Loss: 0.2174 | Time: 51.34s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:57:12] [Round 045] FedProx | Acc: 94.64% | Loss: 0.2150 | Time: 21.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:57:25] [Round 046] FedProx | Acc: 94.63% | Loss: 0.2142 | Time: 13.13s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:57:37] [Round 047] FedProx | Acc: 94.65% | Loss: 0.2134 | Time: 12.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:57:49] [Round 048] FedProx | Acc: 94.76% | Loss: 0.2123 | Time: 11.97s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:58:01] [Round 049] FedProx | Acc: 94.67% | Loss: 0.2089 | Time: 11.97s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:58:13] [Round 050] FedProx | Acc: 94.78% | Loss: 0.2081 | Time: 11.92s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:58:13] Saved to runs/run_20251127_001740_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s2.npz
