[2025-11-27 04:16:53] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', 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_001404_resnet18_head/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.1)
[2025-11-27 04:16:53] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 04:18:40] [Round 001] FedProx | Acc: 16.02% | Loss: 2.9744 | Time: 5.84s | Max GPU Mem: 1506.73 MB
[2025-11-27 04:19:49] [Round 002] FedProx | Acc: 35.65% | Loss: 2.6019 | Time: 69.01s | Max GPU Mem: 1507.90 MB
[2025-11-27 04:21:01] [Round 003] FedProx | Acc: 45.61% | Loss: 2.4119 | Time: 71.88s | Max GPU Mem: 1506.77 MB
[2025-11-27 04:22:21] [Round 004] FedProx | Acc: 54.45% | Loss: 1.3401 | Time: 79.94s | Max GPU Mem: 1507.54 MB
[2025-11-27 04:23:28] [Round 005] FedProx | Acc: 56.10% | Loss: 1.2836 | Time: 67.56s | Max GPU Mem: 1507.42 MB
[2025-11-27 04:24:24] [Round 006] FedProx | Acc: 51.68% | Loss: 1.3227 | Time: 55.73s | Max GPU Mem: 1507.29 MB
[2025-11-27 04:25:54] [Round 007] FedProx | Acc: 51.52% | Loss: 1.3717 | Time: 89.61s | Max GPU Mem: 1507.44 MB
[2025-11-27 04:27:37] [Round 008] FedProx | Acc: 71.48% | Loss: 0.9041 | Time: 103.19s | Max GPU Mem: 1507.56 MB
[2025-11-27 04:28:40] [Round 009] FedProx | Acc: 56.10% | Loss: 1.1461 | Time: 62.90s | Max GPU Mem: 1507.44 MB
[2025-11-27 04:29:42] [Round 010] FedProx | Acc: 53.00% | Loss: 1.4160 | Time: 61.96s | Max GPU Mem: 1507.31 MB
[2025-11-27 04:30:28] [Round 011] FedProx | Acc: 62.70% | Loss: 1.0638 | Time: 46.46s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:31:13] [Round 012] FedProx | Acc: 61.51% | Loss: 1.1935 | Time: 44.81s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:32:08] [Round 013] FedProx | Acc: 63.96% | Loss: 0.9229 | Time: 55.15s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:33:05] [Round 014] FedProx | Acc: 56.17% | Loss: 1.1965 | Time: 57.23s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:33:51] [Round 015] FedProx | Acc: 57.44% | Loss: 1.0643 | Time: 45.64s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:34:51] [Round 016] FedProx | Acc: 63.49% | Loss: 0.8970 | Time: 60.25s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:35:38] [Round 017] FedProx | Acc: 67.12% | Loss: 0.8925 | Time: 46.82s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:36:35] [Round 018] FedProx | Acc: 73.61% | Loss: 0.7244 | Time: 57.07s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:37:37] [Round 019] FedProx | Acc: 74.01% | Loss: 0.6607 | Time: 62.15s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:38:11] [Round 020] FedProx | Acc: 69.16% | Loss: 0.7571 | Time: 33.94s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:39:09] [Round 021] FedProx | Acc: 72.21% | Loss: 0.7421 | Time: 57.70s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:40:16] [Round 022] FedProx | Acc: 78.49% | Loss: 0.6050 | Time: 67.43s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:41:13] [Round 023] FedProx | Acc: 72.64% | Loss: 0.8282 | Time: 56.48s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:42:11] [Round 024] FedProx | Acc: 81.22% | Loss: 0.5514 | Time: 57.99s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:43:54] [Round 025] FedProx | Acc: 77.26% | Loss: 0.6361 | Time: 102.76s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:45:18] [Round 026] FedProx | Acc: 81.59% | Loss: 0.5634 | Time: 84.63s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:46:54] [Round 027] FedProx | Acc: 81.46% | Loss: 0.5648 | Time: 95.63s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:48:34] [Round 028] FedProx | Acc: 82.01% | Loss: 0.5305 | Time: 99.67s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:50:07] [Round 029] FedProx | Acc: 80.09% | Loss: 0.5578 | Time: 93.30s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:51:11] [Round 030] FedProx | Acc: 66.91% | Loss: 1.0181 | Time: 64.10s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:52:31] [Round 031] FedProx | Acc: 80.50% | Loss: 0.5340 | Time: 80.01s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:53:48] [Round 032] FedProx | Acc: 69.34% | Loss: 0.8679 | Time: 77.31s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:55:05] [Round 033] FedProx | Acc: 79.60% | Loss: 0.5417 | Time: 76.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:55:59] [Round 034] FedProx | Acc: 74.13% | Loss: 0.7051 | Time: 54.58s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:58:02] [Round 035] FedProx | Acc: 79.28% | Loss: 0.5492 | Time: 122.99s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:58:58] [Round 036] FedProx | Acc: 74.06% | Loss: 0.7061 | Time: 56.11s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:00:17] [Round 037] FedProx | Acc: 80.85% | Loss: 0.5529 | Time: 79.26s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:01:35] [Round 038] FedProx | Acc: 76.83% | Loss: 0.6683 | Time: 77.08s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:02:30] [Round 039] FedProx | Acc: 78.32% | Loss: 0.5465 | Time: 55.17s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:04:03] [Round 040] FedProx | Acc: 76.21% | Loss: 0.6278 | Time: 93.69s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:04:37] [Round 041] FedProx | Acc: 70.44% | Loss: 0.7417 | Time: 33.95s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:04:55] [Round 042] FedProx | Acc: 73.27% | Loss: 0.7231 | Time: 17.30s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:05:08] [Round 043] FedProx | Acc: 61.49% | Loss: 1.1420 | Time: 13.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:05:32] [Round 044] FedProx | Acc: 80.71% | Loss: 0.5435 | Time: 23.81s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:05:39] [Round 045] FedProx | Acc: 58.48% | Loss: 1.2650 | Time: 7.04s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:05:49] [Round 046] FedProx | Acc: 75.61% | Loss: 0.7622 | Time: 9.86s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:06:04] [Round 047] FedProx | Acc: 73.48% | Loss: 0.6859 | Time: 15.45s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:06:14] [Round 048] FedProx | Acc: 75.65% | Loss: 0.7078 | Time: 10.42s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:06:21] [Round 049] FedProx | Acc: 68.97% | Loss: 1.0055 | Time: 6.65s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:06:32] [Round 050] FedProx | Acc: 76.26% | Loss: 0.6854 | Time: 10.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:06:32] Saved to runs/run_20251127_001404_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s1.npz
