[2025-11-27 04:16:00] 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_001404_resnet18_head/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.1)
[2025-11-27 04:16:00] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 04:18:05] [Round 001] FedProx | Acc: 46.87% | Loss: 1.8384 | Time: 12.89s | Max GPU Mem: 1506.73 MB
[2025-11-27 04:18:15] [Round 002] FedProx | Acc: 41.51% | Loss: 1.8037 | Time: 9.84s | Max GPU Mem: 1507.90 MB
[2025-11-27 04:18:26] [Round 003] FedProx | Acc: 53.02% | Loss: 1.2533 | Time: 10.36s | Max GPU Mem: 1506.81 MB
[2025-11-27 04:18:38] [Round 004] FedProx | Acc: 46.01% | Loss: 1.5704 | Time: 12.11s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:19:27] [Round 005] FedProx | Acc: 67.42% | Loss: 1.0105 | Time: 49.17s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:20:16] [Round 006] FedProx | Acc: 57.89% | Loss: 1.2123 | Time: 49.61s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:21:11] [Round 007] FedProx | Acc: 51.28% | Loss: 1.1848 | Time: 54.76s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:22:19] [Round 008] FedProx | Acc: 67.76% | Loss: 0.9222 | Time: 67.68s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:23:18] [Round 009] FedProx | Acc: 71.59% | Loss: 0.8137 | Time: 59.25s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:24:05] [Round 010] FedProx | Acc: 54.13% | Loss: 1.4787 | Time: 47.20s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:24:52] [Round 011] FedProx | Acc: 66.40% | Loss: 1.0094 | Time: 46.68s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:25:50] [Round 012] FedProx | Acc: 67.68% | Loss: 0.8039 | Time: 58.04s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:26:59] [Round 013] FedProx | Acc: 72.96% | Loss: 0.7634 | Time: 68.69s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:28:05] [Round 014] FedProx | Acc: 73.39% | Loss: 0.7682 | Time: 66.08s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:29:03] [Round 015] FedProx | Acc: 64.78% | Loss: 0.8990 | Time: 58.22s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:30:04] [Round 016] FedProx | Acc: 66.18% | Loss: 0.8524 | Time: 60.85s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:31:14] [Round 017] FedProx | Acc: 62.88% | Loss: 1.1260 | Time: 69.66s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:32:36] [Round 018] FedProx | Acc: 76.88% | Loss: 0.7016 | Time: 82.49s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:33:54] [Round 019] FedProx | Acc: 74.27% | Loss: 0.6906 | Time: 78.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:35:09] [Round 020] FedProx | Acc: 72.00% | Loss: 0.7475 | Time: 74.71s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:36:33] [Round 021] FedProx | Acc: 74.98% | Loss: 0.6933 | Time: 84.04s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:37:47] [Round 022] FedProx | Acc: 70.88% | Loss: 0.7559 | Time: 73.98s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:38:58] [Round 023] FedProx | Acc: 71.89% | Loss: 0.7713 | Time: 70.80s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:40:15] [Round 024] FedProx | Acc: 72.36% | Loss: 0.7412 | Time: 77.60s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:41:31] [Round 025] FedProx | Acc: 78.26% | Loss: 0.6129 | Time: 75.39s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:42:47] [Round 026] FedProx | Acc: 77.87% | Loss: 0.6332 | Time: 76.42s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:43:57] [Round 027] FedProx | Acc: 74.05% | Loss: 0.6763 | Time: 69.59s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:44:57] [Round 028] FedProx | Acc: 75.19% | Loss: 0.6745 | Time: 60.64s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:45:46] [Round 029] FedProx | Acc: 75.93% | Loss: 0.6517 | Time: 48.86s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:46:43] [Round 030] FedProx | Acc: 76.70% | Loss: 0.6002 | Time: 56.70s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:47:38] [Round 031] FedProx | Acc: 70.49% | Loss: 0.7924 | Time: 55.27s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:48:35] [Round 032] FedProx | Acc: 77.87% | Loss: 0.6295 | Time: 57.12s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:49:37] [Round 033] FedProx | Acc: 75.93% | Loss: 0.6465 | Time: 62.01s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:50:30] [Round 034] FedProx | Acc: 81.17% | Loss: 0.5455 | Time: 53.00s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:51:19] [Round 035] FedProx | Acc: 81.98% | Loss: 0.5390 | Time: 49.09s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:52:20] [Round 036] FedProx | Acc: 81.57% | Loss: 0.5287 | Time: 60.31s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:53:14] [Round 037] FedProx | Acc: 70.54% | Loss: 0.7495 | Time: 54.29s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:54:05] [Round 038] FedProx | Acc: 67.20% | Loss: 0.8667 | Time: 51.31s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:55:06] [Round 039] FedProx | Acc: 79.99% | Loss: 0.5573 | Time: 60.96s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:56:02] [Round 040] FedProx | Acc: 76.38% | Loss: 0.6535 | Time: 55.33s | Max GPU Mem: 1507.58 MB
[2025-11-27 04:57:05] [Round 041] FedProx | Acc: 76.94% | Loss: 0.6154 | Time: 62.92s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:58:01] [Round 042] FedProx | Acc: 77.69% | Loss: 0.6028 | Time: 55.96s | Max GPU Mem: 1507.33 MB
[2025-11-27 04:58:45] [Round 043] FedProx | Acc: 80.10% | Loss: 0.5605 | Time: 44.02s | Max GPU Mem: 1507.46 MB
[2025-11-27 04:59:29] [Round 044] FedProx | Acc: 75.75% | Loss: 0.6723 | Time: 44.41s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:00:18] [Round 045] FedProx | Acc: 79.83% | Loss: 0.5497 | Time: 48.69s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:01:14] [Round 046] FedProx | Acc: 77.67% | Loss: 0.6374 | Time: 56.77s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:01:58] [Round 047] FedProx | Acc: 75.28% | Loss: 0.6809 | Time: 43.11s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:02:42] [Round 048] FedProx | Acc: 75.07% | Loss: 0.7010 | Time: 44.73s | Max GPU Mem: 1507.58 MB
[2025-11-27 05:03:35] [Round 049] FedProx | Acc: 80.53% | Loss: 0.5540 | Time: 52.37s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:04:34] [Round 050] FedProx | Acc: 82.19% | Loss: 0.5122 | Time: 59.00s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:04:34] Saved to runs/run_20251127_001404_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s0.npz
