[2025-11-27 10:05:01] Namespace(method='FedProx', training_mode='head', dataset='cifar10', 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/cifar10/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=10.0)
[2025-11-27 10:05:01] Start: FedProx | Mode: head | Dataset: cifar10 | Device: cuda
[2025-11-27 10:08:29] [Round 001] FedProx | Acc: 73.92% | Loss: 1.1486 | Time: 92.11s | Max GPU Mem: 1482.23 MB
[2025-11-27 10:10:03] [Round 002] FedProx | Acc: 77.51% | Loss: 0.8782 | Time: 93.93s | Max GPU Mem: 1482.25 MB
[2025-11-27 10:11:40] [Round 003] FedProx | Acc: 79.05% | Loss: 0.7519 | Time: 96.91s | Max GPU Mem: 1482.29 MB
[2025-11-27 10:13:46] [Round 004] FedProx | Acc: 79.65% | Loss: 0.6937 | Time: 126.33s | Max GPU Mem: 1482.31 MB
[2025-11-27 10:15:15] [Round 005] FedProx | Acc: 79.98% | Loss: 0.6552 | Time: 89.06s | Max GPU Mem: 1482.31 MB
[2025-11-27 10:16:40] [Round 006] FedProx | Acc: 80.35% | Loss: 0.6269 | Time: 85.12s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:18:08] [Round 007] FedProx | Acc: 80.89% | Loss: 0.6036 | Time: 87.94s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:19:29] [Round 008] FedProx | Acc: 81.59% | Loss: 0.5834 | Time: 80.74s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:20:50] [Round 009] FedProx | Acc: 81.83% | Loss: 0.5663 | Time: 80.98s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:22:16] [Round 010] FedProx | Acc: 82.06% | Loss: 0.5552 | Time: 85.95s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:23:39] [Round 011] FedProx | Acc: 82.01% | Loss: 0.5487 | Time: 82.90s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:25:04] [Round 012] FedProx | Acc: 82.41% | Loss: 0.5374 | Time: 85.47s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:26:20] [Round 013] FedProx | Acc: 82.36% | Loss: 0.5311 | Time: 75.82s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:27:44] [Round 014] FedProx | Acc: 82.47% | Loss: 0.5257 | Time: 83.99s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:29:02] [Round 015] FedProx | Acc: 82.61% | Loss: 0.5182 | Time: 78.29s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:30:25] [Round 016] FedProx | Acc: 82.93% | Loss: 0.5111 | Time: 82.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:32:03] [Round 017] FedProx | Acc: 83.18% | Loss: 0.5061 | Time: 98.63s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:34:06] [Round 018] FedProx | Acc: 83.11% | Loss: 0.4988 | Time: 122.58s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:36:27] [Round 019] FedProx | Acc: 83.42% | Loss: 0.4952 | Time: 140.61s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:37:55] [Round 020] FedProx | Acc: 83.57% | Loss: 0.4882 | Time: 88.77s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:39:25] [Round 021] FedProx | Acc: 83.38% | Loss: 0.4913 | Time: 89.35s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:40:40] [Round 022] FedProx | Acc: 83.55% | Loss: 0.4833 | Time: 75.29s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:41:51] [Round 023] FedProx | Acc: 83.57% | Loss: 0.4847 | Time: 71.02s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:43:30] [Round 024] FedProx | Acc: 83.68% | Loss: 0.4814 | Time: 98.68s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:44:58] [Round 025] FedProx | Acc: 83.82% | Loss: 0.4783 | Time: 88.68s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:46:32] [Round 026] FedProx | Acc: 83.90% | Loss: 0.4742 | Time: 93.73s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:47:39] [Round 027] FedProx | Acc: 84.10% | Loss: 0.4678 | Time: 67.07s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:49:14] [Round 028] FedProx | Acc: 84.04% | Loss: 0.4702 | Time: 95.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:50:57] [Round 029] FedProx | Acc: 84.18% | Loss: 0.4663 | Time: 102.34s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:52:30] [Round 030] FedProx | Acc: 83.92% | Loss: 0.4718 | Time: 93.27s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:53:53] [Round 031] FedProx | Acc: 84.57% | Loss: 0.4569 | Time: 83.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:55:15] [Round 032] FedProx | Acc: 84.70% | Loss: 0.4537 | Time: 81.72s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:57:08] [Round 033] FedProx | Acc: 84.55% | Loss: 0.4540 | Time: 112.92s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:58:56] [Round 034] FedProx | Acc: 84.55% | Loss: 0.4532 | Time: 107.38s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:00:22] [Round 035] FedProx | Acc: 84.75% | Loss: 0.4489 | Time: 86.50s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:01:39] [Round 036] FedProx | Acc: 84.58% | Loss: 0.4523 | Time: 77.39s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:03:11] [Round 037] FedProx | Acc: 84.45% | Loss: 0.4520 | Time: 91.89s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:04:38] [Round 038] FedProx | Acc: 84.77% | Loss: 0.4451 | Time: 86.28s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:06:07] [Round 039] FedProx | Acc: 84.89% | Loss: 0.4430 | Time: 89.84s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:08:08] [Round 040] FedProx | Acc: 84.76% | Loss: 0.4473 | Time: 121.04s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:09:58] [Round 041] FedProx | Acc: 84.79% | Loss: 0.4446 | Time: 109.64s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:11:19] [Round 042] FedProx | Acc: 84.97% | Loss: 0.4405 | Time: 80.96s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:11:46] [Round 043] FedProx | Acc: 85.00% | Loss: 0.4382 | Time: 27.11s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:12:12] [Round 044] FedProx | Acc: 84.76% | Loss: 0.4408 | Time: 25.89s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:12:40] [Round 045] FedProx | Acc: 85.00% | Loss: 0.4384 | Time: 27.95s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:13:04] [Round 046] FedProx | Acc: 85.06% | Loss: 0.4371 | Time: 23.77s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:14:44] [Round 047] FedProx | Acc: 85.19% | Loss: 0.4329 | Time: 100.36s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:15:45] [Round 048] FedProx | Acc: 85.48% | Loss: 0.4286 | Time: 61.16s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:16:56] [Round 049] FedProx | Acc: 85.23% | Loss: 0.4304 | Time: 70.72s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:18:09] [Round 050] FedProx | Acc: 85.08% | Loss: 0.4359 | Time: 72.87s | Max GPU Mem: 1482.33 MB
[2025-11-27 11:18:09] Saved to runs/run_20251127_001740_resnet18_head/cifar10/logs/FedProx_head_cifar10_resnet18_s2.npz
