[2025-11-27 05:17:13] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', 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/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 05:17:13] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:20:08] [Round 001] FedProx | Acc: 45.91% | Loss: 1.4482 | Time: 62.84s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:21:26] [Round 002] FedProx | Acc: 71.13% | Loss: 0.9452 | Time: 77.70s | Max GPU Mem: 1507.92 MB
[2025-11-27 05:22:38] [Round 003] FedProx | Acc: 75.48% | Loss: 0.7541 | Time: 72.51s | Max GPU Mem: 1506.79 MB
[2025-11-27 05:24:00] [Round 004] FedProx | Acc: 75.83% | Loss: 0.7010 | Time: 81.77s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:25:01] [Round 005] FedProx | Acc: 71.19% | Loss: 0.7957 | Time: 60.52s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:26:32] [Round 006] FedProx | Acc: 73.95% | Loss: 0.7170 | Time: 91.07s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:27:52] [Round 007] FedProx | Acc: 77.39% | Loss: 0.6369 | Time: 79.94s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:28:59] [Round 008] FedProx | Acc: 78.71% | Loss: 0.6189 | Time: 67.54s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:30:04] [Round 009] FedProx | Acc: 76.03% | Loss: 0.7395 | Time: 64.64s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:31:09] [Round 010] FedProx | Acc: 80.93% | Loss: 0.5608 | Time: 65.69s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:32:05] [Round 011] FedProx | Acc: 81.30% | Loss: 0.5414 | Time: 55.51s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:32:45] [Round 012] FedProx | Acc: 82.55% | Loss: 0.5337 | Time: 39.63s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:33:39] [Round 013] FedProx | Acc: 81.10% | Loss: 0.5560 | Time: 54.30s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:34:34] [Round 014] FedProx | Acc: 75.35% | Loss: 0.6650 | Time: 54.79s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:35:41] [Round 015] FedProx | Acc: 77.03% | Loss: 0.6391 | Time: 66.79s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:36:46] [Round 016] FedProx | Acc: 81.19% | Loss: 0.5319 | Time: 65.48s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:37:48] [Round 017] FedProx | Acc: 81.30% | Loss: 0.5286 | Time: 62.00s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:38:51] [Round 018] FedProx | Acc: 82.05% | Loss: 0.5075 | Time: 63.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:39:43] [Round 019] FedProx | Acc: 76.49% | Loss: 0.6191 | Time: 52.01s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:40:39] [Round 020] FedProx | Acc: 78.87% | Loss: 0.5604 | Time: 55.77s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:41:38] [Round 021] FedProx | Acc: 78.64% | Loss: 0.5880 | Time: 59.52s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:42:32] [Round 022] FedProx | Acc: 81.02% | Loss: 0.5161 | Time: 53.59s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:43:45] [Round 023] FedProx | Acc: 79.62% | Loss: 0.5439 | Time: 72.83s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:44:51] [Round 024] FedProx | Acc: 81.76% | Loss: 0.5005 | Time: 66.00s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:45:42] [Round 025] FedProx | Acc: 84.01% | Loss: 0.4653 | Time: 51.07s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:46:47] [Round 026] FedProx | Acc: 81.87% | Loss: 0.5105 | Time: 65.09s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:48:31] [Round 027] FedProx | Acc: 80.68% | Loss: 0.5207 | Time: 104.53s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:49:31] [Round 028] FedProx | Acc: 77.85% | Loss: 0.6236 | Time: 59.69s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:50:36] [Round 029] FedProx | Acc: 83.36% | Loss: 0.4779 | Time: 65.02s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:51:43] [Round 030] FedProx | Acc: 82.03% | Loss: 0.4880 | Time: 67.01s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:52:40] [Round 031] FedProx | Acc: 81.18% | Loss: 0.5058 | Time: 56.66s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:53:30] [Round 032] FedProx | Acc: 79.09% | Loss: 0.6001 | Time: 50.14s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:54:30] [Round 033] FedProx | Acc: 82.79% | Loss: 0.4773 | Time: 60.22s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:55:09] [Round 034] FedProx | Acc: 83.00% | Loss: 0.4693 | Time: 39.06s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:55:28] [Round 035] FedProx | Acc: 80.09% | Loss: 0.5283 | Time: 18.47s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:55:48] [Round 036] FedProx | Acc: 84.51% | Loss: 0.4387 | Time: 19.99s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:56:10] [Round 037] FedProx | Acc: 81.31% | Loss: 0.5074 | Time: 22.49s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:56:30] [Round 038] FedProx | Acc: 84.83% | Loss: 0.4365 | Time: 20.15s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:56:49] [Round 039] FedProx | Acc: 84.29% | Loss: 0.4467 | Time: 18.52s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:57:26] [Round 040] FedProx | Acc: 80.22% | Loss: 0.5343 | Time: 37.46s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:58:09] [Round 041] FedProx | Acc: 84.02% | Loss: 0.4422 | Time: 42.95s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:59:13] [Round 042] FedProx | Acc: 85.40% | Loss: 0.4250 | Time: 63.35s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:59:38] [Round 043] FedProx | Acc: 83.28% | Loss: 0.4628 | Time: 25.03s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:00:00] [Round 044] FedProx | Acc: 83.60% | Loss: 0.4635 | Time: 22.87s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:00:23] [Round 045] FedProx | Acc: 81.23% | Loss: 0.5048 | Time: 22.51s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:00:45] [Round 046] FedProx | Acc: 81.00% | Loss: 0.5010 | Time: 21.52s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:01:08] [Round 047] FedProx | Acc: 84.16% | Loss: 0.4424 | Time: 23.50s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:01:56] [Round 048] FedProx | Acc: 84.23% | Loss: 0.4424 | Time: 48.16s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:02:58] [Round 049] FedProx | Acc: 83.48% | Loss: 0.4602 | Time: 61.68s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:03:51] [Round 050] FedProx | Acc: 82.98% | Loss: 0.4649 | Time: 53.03s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:03:51] Saved to runs/run_20251127_001608_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s2.npz
