[2025-11-27 10:00:21] Namespace(method='FedProx', training_mode='head', dataset='cifar10', 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/cifar10/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 10:00:21] Start: FedProx | Mode: head | Dataset: cifar10 | Device: cuda
[2025-11-27 10:02:44] [Round 001] FedProx | Acc: 74.77% | Loss: 1.1550 | Time: 28.80s | Max GPU Mem: 1482.23 MB
[2025-11-27 10:04:11] [Round 002] FedProx | Acc: 77.22% | Loss: 0.8794 | Time: 86.63s | Max GPU Mem: 1482.27 MB
[2025-11-27 10:05:04] [Round 003] FedProx | Acc: 79.09% | Loss: 0.7565 | Time: 53.88s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:05:28] [Round 004] FedProx | Acc: 78.81% | Loss: 0.7100 | Time: 24.03s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:05:53] [Round 005] FedProx | Acc: 79.92% | Loss: 0.6609 | Time: 24.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:06:18] [Round 006] FedProx | Acc: 80.63% | Loss: 0.6311 | Time: 25.30s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:06:44] [Round 007] FedProx | Acc: 80.71% | Loss: 0.6120 | Time: 25.72s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:07:42] [Round 008] FedProx | Acc: 81.13% | Loss: 0.5904 | Time: 58.28s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:09:23] [Round 009] FedProx | Acc: 82.04% | Loss: 0.5660 | Time: 101.02s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:10:58] [Round 010] FedProx | Acc: 82.03% | Loss: 0.5574 | Time: 94.58s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:12:51] [Round 011] FedProx | Acc: 82.20% | Loss: 0.5473 | Time: 113.42s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:14:30] [Round 012] FedProx | Acc: 82.46% | Loss: 0.5398 | Time: 98.34s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:16:02] [Round 013] FedProx | Acc: 82.37% | Loss: 0.5327 | Time: 92.07s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:17:24] [Round 014] FedProx | Acc: 82.77% | Loss: 0.5236 | Time: 82.12s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:18:52] [Round 015] FedProx | Acc: 83.04% | Loss: 0.5127 | Time: 87.92s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:20:07] [Round 016] FedProx | Acc: 83.08% | Loss: 0.5087 | Time: 75.55s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:21:32] [Round 017] FedProx | Acc: 83.15% | Loss: 0.5028 | Time: 84.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:22:57] [Round 018] FedProx | Acc: 83.06% | Loss: 0.5040 | Time: 85.85s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:24:23] [Round 019] FedProx | Acc: 83.08% | Loss: 0.5011 | Time: 85.45s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:25:39] [Round 020] FedProx | Acc: 83.38% | Loss: 0.4913 | Time: 75.64s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:27:10] [Round 021] FedProx | Acc: 83.36% | Loss: 0.4899 | Time: 91.60s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:28:45] [Round 022] FedProx | Acc: 83.57% | Loss: 0.4815 | Time: 95.29s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:29:59] [Round 023] FedProx | Acc: 83.69% | Loss: 0.4810 | Time: 73.61s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:31:15] [Round 024] FedProx | Acc: 83.70% | Loss: 0.4816 | Time: 75.68s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:32:57] [Round 025] FedProx | Acc: 84.09% | Loss: 0.4700 | Time: 102.60s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:35:35] [Round 026] FedProx | Acc: 84.15% | Loss: 0.4684 | Time: 157.29s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:37:00] [Round 027] FedProx | Acc: 84.10% | Loss: 0.4697 | Time: 85.02s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:38:15] [Round 028] FedProx | Acc: 84.19% | Loss: 0.4651 | Time: 75.26s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:39:51] [Round 029] FedProx | Acc: 84.21% | Loss: 0.4651 | Time: 96.02s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:41:21] [Round 030] FedProx | Acc: 84.39% | Loss: 0.4610 | Time: 90.44s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:42:31] [Round 031] FedProx | Acc: 84.46% | Loss: 0.4600 | Time: 69.35s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:44:04] [Round 032] FedProx | Acc: 84.51% | Loss: 0.4515 | Time: 93.30s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:45:18] [Round 033] FedProx | Acc: 84.51% | Loss: 0.4564 | Time: 74.12s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:46:42] [Round 034] FedProx | Acc: 84.54% | Loss: 0.4534 | Time: 83.76s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:47:46] [Round 035] FedProx | Acc: 84.63% | Loss: 0.4501 | Time: 63.85s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:49:21] [Round 036] FedProx | Acc: 84.77% | Loss: 0.4456 | Time: 95.30s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:51:01] [Round 037] FedProx | Acc: 84.77% | Loss: 0.4459 | Time: 99.87s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:52:36] [Round 038] FedProx | Acc: 84.98% | Loss: 0.4447 | Time: 95.01s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:53:58] [Round 039] FedProx | Acc: 84.71% | Loss: 0.4491 | Time: 82.02s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:55:16] [Round 040] FedProx | Acc: 84.87% | Loss: 0.4405 | Time: 78.26s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:57:06] [Round 041] FedProx | Acc: 84.87% | Loss: 0.4393 | Time: 109.98s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:58:56] [Round 042] FedProx | Acc: 84.98% | Loss: 0.4361 | Time: 109.55s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:00:28] [Round 043] FedProx | Acc: 85.05% | Loss: 0.4353 | Time: 91.87s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:01:46] [Round 044] FedProx | Acc: 85.10% | Loss: 0.4343 | Time: 78.69s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:03:17] [Round 045] FedProx | Acc: 84.92% | Loss: 0.4378 | Time: 90.38s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:04:40] [Round 046] FedProx | Acc: 85.15% | Loss: 0.4374 | Time: 82.92s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:06:06] [Round 047] FedProx | Acc: 85.17% | Loss: 0.4335 | Time: 86.83s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:08:07] [Round 048] FedProx | Acc: 85.11% | Loss: 0.4346 | Time: 120.28s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:09:54] [Round 049] FedProx | Acc: 85.23% | Loss: 0.4324 | Time: 107.73s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:11:18] [Round 050] FedProx | Acc: 85.19% | Loss: 0.4317 | Time: 83.17s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:11:18] Saved to runs/run_20251127_001740_resnet18_head/cifar10/logs/FedProx_head_cifar10_resnet18_s0.npz
