[2025-11-27 09:15:25] 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_001608_resnet18_head/cifar10/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 09:15:25] Start: FedProx | Mode: head | Dataset: cifar10 | Device: cuda
[2025-11-27 09:17:37] [Round 001] FedProx | Acc: 44.38% | Loss: 1.6018 | Time: 18.03s | Max GPU Mem: 1482.23 MB
[2025-11-27 09:17:55] [Round 002] FedProx | Acc: 68.80% | Loss: 1.0977 | Time: 18.11s | Max GPU Mem: 1482.29 MB
[2025-11-27 09:18:15] [Round 003] FedProx | Acc: 66.73% | Loss: 0.9895 | Time: 20.73s | Max GPU Mem: 1482.29 MB
[2025-11-27 09:18:39] [Round 004] FedProx | Acc: 67.84% | Loss: 0.9345 | Time: 23.67s | Max GPU Mem: 1482.31 MB
[2025-11-27 09:19:19] [Round 005] FedProx | Acc: 71.10% | Loss: 0.8589 | Time: 39.91s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:20:20] [Round 006] FedProx | Acc: 75.56% | Loss: 0.7375 | Time: 60.73s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:21:40] [Round 007] FedProx | Acc: 75.31% | Loss: 0.7311 | Time: 79.80s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:22:42] [Round 008] FedProx | Acc: 77.15% | Loss: 0.6895 | Time: 62.86s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:24:02] [Round 009] FedProx | Acc: 77.74% | Loss: 0.6656 | Time: 79.25s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:24:57] [Round 010] FedProx | Acc: 75.93% | Loss: 0.6865 | Time: 55.59s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:26:06] [Round 011] FedProx | Acc: 76.95% | Loss: 0.6657 | Time: 68.40s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:27:19] [Round 012] FedProx | Acc: 80.40% | Loss: 0.5927 | Time: 73.79s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:28:21] [Round 013] FedProx | Acc: 80.45% | Loss: 0.5888 | Time: 61.27s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:29:27] [Round 014] FedProx | Acc: 79.95% | Loss: 0.5911 | Time: 65.82s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:30:12] [Round 015] FedProx | Acc: 79.52% | Loss: 0.5973 | Time: 45.08s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:30:55] [Round 016] FedProx | Acc: 81.15% | Loss: 0.5665 | Time: 43.51s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:31:47] [Round 017] FedProx | Acc: 79.86% | Loss: 0.5846 | Time: 52.09s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:32:34] [Round 018] FedProx | Acc: 78.91% | Loss: 0.6053 | Time: 47.12s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:33:24] [Round 019] FedProx | Acc: 82.09% | Loss: 0.5324 | Time: 49.99s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:34:15] [Round 020] FedProx | Acc: 80.67% | Loss: 0.5675 | Time: 51.18s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:34:58] [Round 021] FedProx | Acc: 81.66% | Loss: 0.5359 | Time: 42.13s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:35:58] [Round 022] FedProx | Acc: 80.95% | Loss: 0.5517 | Time: 59.91s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:36:56] [Round 023] FedProx | Acc: 81.27% | Loss: 0.5456 | Time: 58.69s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:37:51] [Round 024] FedProx | Acc: 81.18% | Loss: 0.5461 | Time: 54.74s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:38:40] [Round 025] FedProx | Acc: 82.70% | Loss: 0.5045 | Time: 48.85s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:39:24] [Round 026] FedProx | Acc: 79.87% | Loss: 0.5759 | Time: 44.57s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:40:22] [Round 027] FedProx | Acc: 82.49% | Loss: 0.5115 | Time: 57.45s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:41:15] [Round 028] FedProx | Acc: 81.98% | Loss: 0.5232 | Time: 53.01s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:42:20] [Round 029] FedProx | Acc: 81.27% | Loss: 0.5341 | Time: 65.49s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:43:11] [Round 030] FedProx | Acc: 82.97% | Loss: 0.4968 | Time: 51.15s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:44:06] [Round 031] FedProx | Acc: 81.85% | Loss: 0.5255 | Time: 54.21s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:44:54] [Round 032] FedProx | Acc: 81.39% | Loss: 0.5327 | Time: 48.17s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:45:51] [Round 033] FedProx | Acc: 82.08% | Loss: 0.5096 | Time: 56.65s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:46:57] [Round 034] FedProx | Acc: 83.67% | Loss: 0.4762 | Time: 66.11s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:47:55] [Round 035] FedProx | Acc: 82.64% | Loss: 0.5000 | Time: 58.63s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:48:47] [Round 036] FedProx | Acc: 81.08% | Loss: 0.5337 | Time: 51.96s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:49:40] [Round 037] FedProx | Acc: 83.93% | Loss: 0.4674 | Time: 52.97s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:50:33] [Round 038] FedProx | Acc: 83.21% | Loss: 0.4851 | Time: 53.11s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:51:39] [Round 039] FedProx | Acc: 82.71% | Loss: 0.4989 | Time: 65.30s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:52:48] [Round 040] FedProx | Acc: 82.30% | Loss: 0.5014 | Time: 69.20s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:53:48] [Round 041] FedProx | Acc: 83.55% | Loss: 0.4755 | Time: 60.49s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:54:48] [Round 042] FedProx | Acc: 84.35% | Loss: 0.4577 | Time: 59.27s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:55:41] [Round 043] FedProx | Acc: 82.07% | Loss: 0.5068 | Time: 53.49s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:56:49] [Round 044] FedProx | Acc: 83.06% | Loss: 0.4840 | Time: 68.35s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:57:47] [Round 045] FedProx | Acc: 84.43% | Loss: 0.4526 | Time: 57.80s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:58:32] [Round 046] FedProx | Acc: 83.85% | Loss: 0.4627 | Time: 44.96s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:59:26] [Round 047] FedProx | Acc: 83.71% | Loss: 0.4674 | Time: 53.54s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:00:22] [Round 048] FedProx | Acc: 82.16% | Loss: 0.5068 | Time: 56.19s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:01:17] [Round 049] FedProx | Acc: 83.72% | Loss: 0.4649 | Time: 54.62s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:02:04] [Round 050] FedProx | Acc: 83.04% | Loss: 0.4821 | Time: 47.90s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:02:04] Saved to runs/run_20251127_001608_resnet18_head/cifar10/logs/FedProx_head_cifar10_resnet18_s0.npz
