[2025-11-27 05:16:16] Namespace(method='FedProx', training_mode='head', dataset='fashionmnist', model_name='resnet18', seed=1, 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/fashionmnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 05:16:16] Start: FedProx | Mode: head | Dataset: fashionmnist | Device: cuda
[2025-11-27 05:18:26] [Round 001] FedProx | Acc: 53.99% | Loss: 1.3873 | Time: 17.18s | Max GPU Mem: 1506.73 MB
[2025-11-27 05:18:48] [Round 002] FedProx | Acc: 66.64% | Loss: 1.0270 | Time: 21.74s | Max GPU Mem: 1507.92 MB
[2025-11-27 05:19:05] [Round 003] FedProx | Acc: 78.93% | Loss: 0.8019 | Time: 16.84s | Max GPU Mem: 1506.81 MB
[2025-11-27 05:19:52] [Round 004] FedProx | Acc: 71.59% | Loss: 0.8223 | Time: 47.17s | Max GPU Mem: 1507.06 MB
[2025-11-27 05:20:39] [Round 005] FedProx | Acc: 75.42% | Loss: 0.7275 | Time: 46.78s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:21:23] [Round 006] FedProx | Acc: 79.36% | Loss: 0.6510 | Time: 44.89s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:22:08] [Round 007] FedProx | Acc: 76.92% | Loss: 0.6764 | Time: 44.61s | Max GPU Mem: 1507.46 MB
[2025-11-27 05:23:09] [Round 008] FedProx | Acc: 80.05% | Loss: 0.6078 | Time: 60.59s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:24:02] [Round 009] FedProx | Acc: 78.65% | Loss: 0.6279 | Time: 52.89s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:24:59] [Round 010] FedProx | Acc: 81.44% | Loss: 0.5669 | Time: 57.03s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:25:41] [Round 011] FedProx | Acc: 81.09% | Loss: 0.5700 | Time: 42.50s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:26:30] [Round 012] FedProx | Acc: 80.92% | Loss: 0.5733 | Time: 49.20s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:27:18] [Round 013] FedProx | Acc: 80.98% | Loss: 0.5586 | Time: 47.86s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:28:05] [Round 014] FedProx | Acc: 80.58% | Loss: 0.5634 | Time: 46.83s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:29:00] [Round 015] FedProx | Acc: 79.92% | Loss: 0.5756 | Time: 55.18s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:29:59] [Round 016] FedProx | Acc: 82.98% | Loss: 0.5081 | Time: 58.40s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:31:07] [Round 017] FedProx | Acc: 83.18% | Loss: 0.5067 | Time: 68.50s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:32:11] [Round 018] FedProx | Acc: 82.96% | Loss: 0.5099 | Time: 63.72s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:33:07] [Round 019] FedProx | Acc: 81.55% | Loss: 0.5361 | Time: 56.18s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:34:12] [Round 020] FedProx | Acc: 81.63% | Loss: 0.5373 | Time: 64.81s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:35:10] [Round 021] FedProx | Acc: 82.61% | Loss: 0.5086 | Time: 57.72s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:36:12] [Round 022] FedProx | Acc: 81.12% | Loss: 0.5404 | Time: 62.73s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:37:12] [Round 023] FedProx | Acc: 82.17% | Loss: 0.5161 | Time: 60.18s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:38:02] [Round 024] FedProx | Acc: 83.48% | Loss: 0.4781 | Time: 49.48s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:39:04] [Round 025] FedProx | Acc: 82.36% | Loss: 0.5158 | Time: 62.10s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:39:53] [Round 026] FedProx | Acc: 83.38% | Loss: 0.4903 | Time: 48.57s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:40:40] [Round 027] FedProx | Acc: 82.89% | Loss: 0.4925 | Time: 47.34s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:41:39] [Round 028] FedProx | Acc: 82.81% | Loss: 0.4973 | Time: 58.97s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:42:31] [Round 029] FedProx | Acc: 83.65% | Loss: 0.4670 | Time: 51.63s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:43:43] [Round 030] FedProx | Acc: 82.67% | Loss: 0.5089 | Time: 72.97s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:44:53] [Round 031] FedProx | Acc: 83.55% | Loss: 0.4810 | Time: 69.11s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:45:44] [Round 032] FedProx | Acc: 84.31% | Loss: 0.4539 | Time: 51.57s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:46:45] [Round 033] FedProx | Acc: 82.75% | Loss: 0.4939 | Time: 60.77s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:47:45] [Round 034] FedProx | Acc: 84.07% | Loss: 0.4621 | Time: 60.20s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:48:38] [Round 035] FedProx | Acc: 83.82% | Loss: 0.4714 | Time: 52.47s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:49:23] [Round 036] FedProx | Acc: 84.94% | Loss: 0.4393 | Time: 45.47s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:50:11] [Round 037] FedProx | Acc: 84.55% | Loss: 0.4507 | Time: 47.79s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:51:15] [Round 038] FedProx | Acc: 83.23% | Loss: 0.4858 | Time: 63.83s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:52:09] [Round 039] FedProx | Acc: 84.76% | Loss: 0.4414 | Time: 54.20s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:53:14] [Round 040] FedProx | Acc: 82.52% | Loss: 0.4929 | Time: 65.59s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:54:31] [Round 041] FedProx | Acc: 84.96% | Loss: 0.4402 | Time: 76.58s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:55:11] [Round 042] FedProx | Acc: 84.81% | Loss: 0.4397 | Time: 40.19s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:55:35] [Round 043] FedProx | Acc: 84.65% | Loss: 0.4457 | Time: 24.10s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:55:53] [Round 044] FedProx | Acc: 85.18% | Loss: 0.4359 | Time: 17.71s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:56:11] [Round 045] FedProx | Acc: 85.14% | Loss: 0.4386 | Time: 17.50s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:56:32] [Round 046] FedProx | Acc: 83.88% | Loss: 0.4582 | Time: 21.41s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:56:50] [Round 047] FedProx | Acc: 83.94% | Loss: 0.4594 | Time: 18.47s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:57:29] [Round 048] FedProx | Acc: 84.70% | Loss: 0.4393 | Time: 38.26s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:58:26] [Round 049] FedProx | Acc: 85.53% | Loss: 0.4204 | Time: 56.87s | Max GPU Mem: 1506.83 MB
[2025-11-27 05:59:17] [Round 050] FedProx | Acc: 84.78% | Loss: 0.4329 | Time: 51.21s | Max GPU Mem: 1507.33 MB
[2025-11-27 05:59:17] Saved to runs/run_20251127_001608_resnet18_head/fashionmnist/logs/FedProx_head_fashionmnist_resnet18_s1.npz
