[2025-11-27 05:26:23] 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_001740_resnet18_head/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 05:26:23] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:29:29] [Round 001] FedProx | Acc: 78.11% | Loss: 0.9302 | Time: 74.78s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:30:42] [Round 002] FedProx | Acc: 79.94% | Loss: 0.7330 | Time: 72.33s | Max GPU Mem: 1507.92 MB
[2025-11-27 05:31:59] [Round 003] FedProx | Acc: 80.87% | Loss: 0.6518 | Time: 77.68s | Max GPU Mem: 1506.81 MB
[2025-11-27 05:33:11] [Round 004] FedProx | Acc: 81.66% | Loss: 0.6084 | Time: 71.39s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:34:21] [Round 005] FedProx | Acc: 81.81% | Loss: 0.5816 | Time: 69.98s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:35:27] [Round 006] FedProx | Acc: 82.39% | Loss: 0.5571 | Time: 66.21s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:36:39] [Round 007] FedProx | Acc: 82.85% | Loss: 0.5395 | Time: 72.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:37:51] [Round 008] FedProx | Acc: 82.94% | Loss: 0.5263 | Time: 71.37s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:39:02] [Round 009] FedProx | Acc: 83.12% | Loss: 0.5151 | Time: 71.09s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:40:10] [Round 010] FedProx | Acc: 83.50% | Loss: 0.5028 | Time: 68.04s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:41:14] [Round 011] FedProx | Acc: 83.50% | Loss: 0.4956 | Time: 64.26s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:42:16] [Round 012] FedProx | Acc: 83.81% | Loss: 0.4865 | Time: 61.81s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:43:17] [Round 013] FedProx | Acc: 84.07% | Loss: 0.4786 | Time: 61.19s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:44:23] [Round 014] FedProx | Acc: 84.17% | Loss: 0.4747 | Time: 65.99s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:45:17] [Round 015] FedProx | Acc: 84.41% | Loss: 0.4672 | Time: 53.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:46:10] [Round 016] FedProx | Acc: 84.47% | Loss: 0.4620 | Time: 53.37s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:47:05] [Round 017] FedProx | Acc: 84.66% | Loss: 0.4576 | Time: 54.50s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:47:59] [Round 018] FedProx | Acc: 84.76% | Loss: 0.4534 | Time: 54.51s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:48:53] [Round 019] FedProx | Acc: 84.70% | Loss: 0.4491 | Time: 53.90s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:49:46] [Round 020] FedProx | Acc: 84.95% | Loss: 0.4477 | Time: 52.59s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:50:40] [Round 021] FedProx | Acc: 84.95% | Loss: 0.4447 | Time: 53.92s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:51:42] [Round 022] FedProx | Acc: 85.16% | Loss: 0.4400 | Time: 62.07s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:52:39] [Round 023] FedProx | Acc: 85.26% | Loss: 0.4358 | Time: 57.79s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:53:34] [Round 024] FedProx | Acc: 85.13% | Loss: 0.4328 | Time: 54.80s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:54:29] [Round 025] FedProx | Acc: 85.27% | Loss: 0.4304 | Time: 54.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:55:34] [Round 026] FedProx | Acc: 85.22% | Loss: 0.4271 | Time: 64.62s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:56:36] [Round 027] FedProx | Acc: 85.47% | Loss: 0.4258 | Time: 62.31s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:57:32] [Round 028] FedProx | Acc: 85.61% | Loss: 0.4239 | Time: 56.49s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:58:28] [Round 029] FedProx | Acc: 85.53% | Loss: 0.4212 | Time: 56.02s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:59:28] [Round 030] FedProx | Acc: 85.69% | Loss: 0.4188 | Time: 59.47s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:00:32] [Round 031] FedProx | Acc: 85.64% | Loss: 0.4164 | Time: 63.72s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:01:27] [Round 032] FedProx | Acc: 85.69% | Loss: 0.4152 | Time: 55.03s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:02:26] [Round 033] FedProx | Acc: 85.74% | Loss: 0.4131 | Time: 59.21s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:03:37] [Round 034] FedProx | Acc: 85.75% | Loss: 0.4129 | Time: 70.87s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:04:46] [Round 035] FedProx | Acc: 85.80% | Loss: 0.4100 | Time: 69.16s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:05:49] [Round 036] FedProx | Acc: 85.83% | Loss: 0.4077 | Time: 62.70s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:06:52] [Round 037] FedProx | Acc: 85.99% | Loss: 0.4067 | Time: 63.20s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:07:48] [Round 038] FedProx | Acc: 85.97% | Loss: 0.4076 | Time: 55.88s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:08:49] [Round 039] FedProx | Acc: 86.11% | Loss: 0.4034 | Time: 61.40s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:09:27] [Round 040] FedProx | Acc: 86.25% | Loss: 0.4041 | Time: 38.05s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:09:50] [Round 041] FedProx | Acc: 86.20% | Loss: 0.4007 | Time: 23.36s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:10:11] [Round 042] FedProx | Acc: 86.28% | Loss: 0.3998 | Time: 20.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:10:34] [Round 043] FedProx | Acc: 86.22% | Loss: 0.4010 | Time: 22.88s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:10:54] [Round 044] FedProx | Acc: 86.47% | Loss: 0.3978 | Time: 19.50s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:11:16] [Round 045] FedProx | Acc: 86.37% | Loss: 0.3951 | Time: 22.86s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:12:19] [Round 046] FedProx | Acc: 86.42% | Loss: 0.3957 | Time: 63.10s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:12:58] [Round 047] FedProx | Acc: 86.43% | Loss: 0.3946 | Time: 38.55s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:13:19] [Round 048] FedProx | Acc: 86.53% | Loss: 0.3937 | Time: 20.75s | Max GPU Mem: 1507.58 MB
[2025-11-27 06:13:41] [Round 049] FedProx | Acc: 86.44% | Loss: 0.3919 | Time: 21.89s | Max GPU Mem: 1507.46 MB
[2025-11-27 06:14:02] [Round 050] FedProx | Acc: 86.52% | Loss: 0.3910 | Time: 21.10s | Max GPU Mem: 1507.33 MB
[2025-11-27 06:14:02] Saved to runs/run_20251127_001740_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s2.npz
