[2025-11-27 05:25:12] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', model_name='resnet18', seed=0, 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/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 05:25:12] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:26:29] [Round 001] FedProx | Acc: 78.01% | Loss: 0.9744 | Time: 13.33s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:26:41] [Round 002] FedProx | Acc: 80.10% | Loss: 0.7401 | Time: 11.93s | Max GPU Mem: 1507.92 MB
[2025-11-27 05:26:53] [Round 003] FedProx | Acc: 80.14% | Loss: 0.6550 | Time: 12.16s | Max GPU Mem: 1506.81 MB
[2025-11-27 05:27:05] [Round 004] FedProx | Acc: 81.30% | Loss: 0.6070 | Time: 11.43s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:27:16] [Round 005] FedProx | Acc: 82.14% | Loss: 0.5735 | Time: 11.64s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:27:28] [Round 006] FedProx | Acc: 82.29% | Loss: 0.5534 | Time: 11.51s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:27:39] [Round 007] FedProx | Acc: 82.81% | Loss: 0.5369 | Time: 11.74s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:27:51] [Round 008] FedProx | Acc: 83.25% | Loss: 0.5211 | Time: 11.63s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:28:03] [Round 009] FedProx | Acc: 83.46% | Loss: 0.5115 | Time: 12.32s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:28:40] [Round 010] FedProx | Acc: 83.52% | Loss: 0.4999 | Time: 36.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:29:32] [Round 011] FedProx | Acc: 83.76% | Loss: 0.4911 | Time: 51.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:30:30] [Round 012] FedProx | Acc: 83.83% | Loss: 0.4854 | Time: 57.76s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:31:12] [Round 013] FedProx | Acc: 84.16% | Loss: 0.4764 | Time: 41.63s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:31:59] [Round 014] FedProx | Acc: 83.98% | Loss: 0.4720 | Time: 47.78s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:32:51] [Round 015] FedProx | Acc: 84.21% | Loss: 0.4670 | Time: 51.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:33:31] [Round 016] FedProx | Acc: 84.43% | Loss: 0.4608 | Time: 40.17s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:34:22] [Round 017] FedProx | Acc: 84.59% | Loss: 0.4574 | Time: 50.57s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:35:15] [Round 018] FedProx | Acc: 84.73% | Loss: 0.4533 | Time: 52.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:36:03] [Round 019] FedProx | Acc: 84.75% | Loss: 0.4492 | Time: 48.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:36:51] [Round 020] FedProx | Acc: 84.90% | Loss: 0.4478 | Time: 47.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:37:47] [Round 021] FedProx | Acc: 85.00% | Loss: 0.4447 | Time: 56.20s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:38:33] [Round 022] FedProx | Acc: 85.01% | Loss: 0.4410 | Time: 46.27s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:39:14] [Round 023] FedProx | Acc: 85.09% | Loss: 0.4382 | Time: 40.51s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:40:13] [Round 024] FedProx | Acc: 85.26% | Loss: 0.4333 | Time: 59.26s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:41:16] [Round 025] FedProx | Acc: 85.34% | Loss: 0.4310 | Time: 63.03s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:42:11] [Round 026] FedProx | Acc: 85.34% | Loss: 0.4282 | Time: 55.44s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:43:03] [Round 027] FedProx | Acc: 85.50% | Loss: 0.4273 | Time: 51.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:43:56] [Round 028] FedProx | Acc: 85.50% | Loss: 0.4234 | Time: 53.48s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:45:02] [Round 029] FedProx | Acc: 85.35% | Loss: 0.4225 | Time: 65.17s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:46:03] [Round 030] FedProx | Acc: 85.58% | Loss: 0.4207 | Time: 61.80s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:47:09] [Round 031] FedProx | Acc: 85.80% | Loss: 0.4179 | Time: 65.24s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:48:11] [Round 032] FedProx | Acc: 85.68% | Loss: 0.4163 | Time: 61.98s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:49:25] [Round 033] FedProx | Acc: 85.74% | Loss: 0.4137 | Time: 74.20s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:50:39] [Round 034] FedProx | Acc: 85.84% | Loss: 0.4114 | Time: 74.57s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:51:44] [Round 035] FedProx | Acc: 85.85% | Loss: 0.4106 | Time: 64.14s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:53:08] [Round 036] FedProx | Acc: 85.97% | Loss: 0.4102 | Time: 84.10s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:54:25] [Round 037] FedProx | Acc: 85.94% | Loss: 0.4062 | Time: 77.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:55:30] [Round 038] FedProx | Acc: 86.00% | Loss: 0.4057 | Time: 64.92s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:56:38] [Round 039] FedProx | Acc: 85.88% | Loss: 0.4072 | Time: 67.43s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:57:40] [Round 040] FedProx | Acc: 86.05% | Loss: 0.4030 | Time: 62.31s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:58:47] [Round 041] FedProx | Acc: 86.05% | Loss: 0.4021 | Time: 67.13s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:00:00] [Round 042] FedProx | Acc: 86.28% | Loss: 0.4021 | Time: 72.50s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:01:19] [Round 043] FedProx | Acc: 86.17% | Loss: 0.3981 | Time: 79.20s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:02:24] [Round 044] FedProx | Acc: 86.18% | Loss: 0.3984 | Time: 65.66s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:03:38] [Round 045] FedProx | Acc: 86.36% | Loss: 0.3974 | Time: 73.41s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:04:52] [Round 046] FedProx | Acc: 86.38% | Loss: 0.3956 | Time: 73.61s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:06:03] [Round 047] FedProx | Acc: 86.30% | Loss: 0.3961 | Time: 71.92s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:07:07] [Round 048] FedProx | Acc: 86.24% | Loss: 0.3943 | Time: 63.33s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:08:12] [Round 049] FedProx | Acc: 86.40% | Loss: 0.3912 | Time: 65.38s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:09:17] [Round 050] FedProx | Acc: 86.48% | Loss: 0.3910 | Time: 64.56s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:09:17] Saved to runs/run_20251127_001740_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s0.npz
