[2025-11-27 01:07:02] 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_001608_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 01:07:02] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:08:35] [Round 001] FedProx | Acc: 57.11% | Loss: 1.5736 | Time: 20.69s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:09:34] [Round 002] FedProx | Acc: 73.17% | Loss: 1.1020 | Time: 59.41s | Max GPU Mem: 1507.90 MB
[2025-11-27 01:10:30] [Round 003] FedProx | Acc: 76.96% | Loss: 0.8919 | Time: 56.07s | Max GPU Mem: 1506.81 MB
[2025-11-27 01:11:27] [Round 004] FedProx | Acc: 86.44% | Loss: 0.6802 | Time: 57.33s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:12:16] [Round 005] FedProx | Acc: 83.45% | Loss: 0.6438 | Time: 48.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:12:58] [Round 006] FedProx | Acc: 85.27% | Loss: 0.5877 | Time: 42.27s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:13:46] [Round 007] FedProx | Acc: 88.38% | Loss: 0.5313 | Time: 48.54s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:14:35] [Round 008] FedProx | Acc: 90.31% | Loss: 0.4854 | Time: 48.35s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:15:27] [Round 009] FedProx | Acc: 88.27% | Loss: 0.4802 | Time: 51.75s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:16:08] [Round 010] FedProx | Acc: 88.68% | Loss: 0.4810 | Time: 41.52s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:16:51] [Round 011] FedProx | Acc: 89.22% | Loss: 0.4604 | Time: 43.05s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:17:30] [Round 012] FedProx | Acc: 88.52% | Loss: 0.4642 | Time: 38.43s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:18:13] [Round 013] FedProx | Acc: 90.59% | Loss: 0.4124 | Time: 43.91s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:56] [Round 014] FedProx | Acc: 90.47% | Loss: 0.3965 | Time: 43.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:19:53] [Round 015] FedProx | Acc: 91.19% | Loss: 0.3805 | Time: 56.73s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:20:46] [Round 016] FedProx | Acc: 90.86% | Loss: 0.3859 | Time: 52.76s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:21:38] [Round 017] FedProx | Acc: 90.81% | Loss: 0.3641 | Time: 52.41s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:22:45] [Round 018] FedProx | Acc: 91.62% | Loss: 0.3471 | Time: 66.72s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:24:05] [Round 019] FedProx | Acc: 92.72% | Loss: 0.3329 | Time: 79.79s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:25:07] [Round 020] FedProx | Acc: 92.07% | Loss: 0.3369 | Time: 61.82s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:26:15] [Round 021] FedProx | Acc: 92.30% | Loss: 0.3298 | Time: 68.28s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:27:16] [Round 022] FedProx | Acc: 92.14% | Loss: 0.3291 | Time: 60.79s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:28:28] [Round 023] FedProx | Acc: 91.67% | Loss: 0.3356 | Time: 71.82s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:29:37] [Round 024] FedProx | Acc: 92.55% | Loss: 0.3093 | Time: 69.20s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:30:32] [Round 025] FedProx | Acc: 92.50% | Loss: 0.3130 | Time: 55.51s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:31:38] [Round 026] FedProx | Acc: 92.80% | Loss: 0.3021 | Time: 65.97s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:32:46] [Round 027] FedProx | Acc: 91.65% | Loss: 0.3064 | Time: 67.57s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:34:18] [Round 028] FedProx | Acc: 93.03% | Loss: 0.2959 | Time: 91.94s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:35:25] [Round 029] FedProx | Acc: 92.54% | Loss: 0.3096 | Time: 67.03s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:36:30] [Round 030] FedProx | Acc: 93.05% | Loss: 0.2881 | Time: 64.88s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:37:36] [Round 031] FedProx | Acc: 93.02% | Loss: 0.2873 | Time: 66.79s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:38:45] [Round 032] FedProx | Acc: 93.17% | Loss: 0.2761 | Time: 68.81s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:39:44] [Round 033] FedProx | Acc: 93.28% | Loss: 0.2694 | Time: 58.88s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:43] [Round 034] FedProx | Acc: 93.05% | Loss: 0.2726 | Time: 58.86s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:41:36] [Round 035] FedProx | Acc: 92.31% | Loss: 0.2829 | Time: 53.32s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:42:32] [Round 036] FedProx | Acc: 92.47% | Loss: 0.2741 | Time: 55.13s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:43:25] [Round 037] FedProx | Acc: 93.24% | Loss: 0.2724 | Time: 53.06s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:44:24] [Round 038] FedProx | Acc: 93.50% | Loss: 0.2598 | Time: 58.94s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:45:18] [Round 039] FedProx | Acc: 92.16% | Loss: 0.2778 | Time: 54.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:46:09] [Round 040] FedProx | Acc: 92.75% | Loss: 0.2829 | Time: 50.75s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:47:10] [Round 041] FedProx | Acc: 91.90% | Loss: 0.2737 | Time: 60.95s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:58] [Round 042] FedProx | Acc: 93.80% | Loss: 0.2453 | Time: 48.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:48:56] [Round 043] FedProx | Acc: 94.09% | Loss: 0.2444 | Time: 58.78s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:52] [Round 044] FedProx | Acc: 92.43% | Loss: 0.2682 | Time: 56.03s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:50:43] [Round 045] FedProx | Acc: 93.47% | Loss: 0.2537 | Time: 50.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:30] [Round 046] FedProx | Acc: 93.32% | Loss: 0.2546 | Time: 47.28s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:52:20] [Round 047] FedProx | Acc: 93.95% | Loss: 0.2407 | Time: 50.50s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:53:12] [Round 048] FedProx | Acc: 92.76% | Loss: 0.2524 | Time: 52.19s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:54:08] [Round 049] FedProx | Acc: 94.04% | Loss: 0.2357 | Time: 55.09s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:55:01] [Round 050] FedProx | Acc: 93.21% | Loss: 0.2460 | Time: 53.87s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:55:01] Saved to runs/run_20251127_001608_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s2.npz
