[2025-11-27 01:02:54] Namespace(method='FedProx', training_mode='head', dataset='mnist', 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_001404_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.1)
[2025-11-27 01:02:54] Start: FedProx | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 01:05:54] [Round 001] FedProx | Acc: 38.30% | Loss: 1.7668 | Time: 68.18s | Max GPU Mem: 1506.73 MB
[2025-11-27 01:07:18] [Round 002] FedProx | Acc: 44.69% | Loss: 1.4669 | Time: 84.39s | Max GPU Mem: 1507.90 MB
[2025-11-27 01:08:25] [Round 003] FedProx | Acc: 41.94% | Loss: 1.7482 | Time: 66.06s | Max GPU Mem: 1506.81 MB
[2025-11-27 01:09:27] [Round 004] FedProx | Acc: 46.29% | Loss: 1.4089 | Time: 62.01s | Max GPU Mem: 1507.56 MB
[2025-11-27 01:11:01] [Round 005] FedProx | Acc: 55.82% | Loss: 1.3073 | Time: 94.66s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:12:40] [Round 006] FedProx | Acc: 71.94% | Loss: 0.8870 | Time: 98.46s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:13:52] [Round 007] FedProx | Acc: 70.67% | Loss: 0.9040 | Time: 72.68s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:15:16] [Round 008] FedProx | Acc: 81.02% | Loss: 0.6897 | Time: 83.40s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:16:22] [Round 009] FedProx | Acc: 69.31% | Loss: 0.9193 | Time: 66.71s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:17:21] [Round 010] FedProx | Acc: 78.38% | Loss: 0.7344 | Time: 58.55s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:18:03] [Round 011] FedProx | Acc: 74.02% | Loss: 0.7132 | Time: 42.12s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:18:51] [Round 012] FedProx | Acc: 86.79% | Loss: 0.5661 | Time: 48.20s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:19:40] [Round 013] FedProx | Acc: 70.20% | Loss: 0.9761 | Time: 48.64s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:20:40] [Round 014] FedProx | Acc: 72.13% | Loss: 0.7889 | Time: 60.00s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:21:40] [Round 015] FedProx | Acc: 81.37% | Loss: 0.5939 | Time: 59.65s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:22:36] [Round 016] FedProx | Acc: 74.35% | Loss: 0.7563 | Time: 55.97s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:23:31] [Round 017] FedProx | Acc: 74.17% | Loss: 0.7500 | Time: 55.36s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:24:07] [Round 018] FedProx | Acc: 66.17% | Loss: 0.8736 | Time: 35.99s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:24:41] [Round 019] FedProx | Acc: 65.97% | Loss: 0.8906 | Time: 33.87s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:25:33] [Round 020] FedProx | Acc: 79.15% | Loss: 0.6054 | Time: 52.30s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:26:19] [Round 021] FedProx | Acc: 68.98% | Loss: 0.8533 | Time: 46.33s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:27:26] [Round 022] FedProx | Acc: 80.03% | Loss: 0.5793 | Time: 66.27s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:28:35] [Round 023] FedProx | Acc: 84.16% | Loss: 0.5135 | Time: 68.92s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:29:42] [Round 024] FedProx | Acc: 78.62% | Loss: 0.5734 | Time: 67.72s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:30:42] [Round 025] FedProx | Acc: 77.55% | Loss: 0.6414 | Time: 59.30s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:31:34] [Round 026] FedProx | Acc: 78.31% | Loss: 0.6234 | Time: 52.27s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:32:39] [Round 027] FedProx | Acc: 81.78% | Loss: 0.5732 | Time: 64.84s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:33:42] [Round 028] FedProx | Acc: 83.59% | Loss: 0.5070 | Time: 63.07s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:34:44] [Round 029] FedProx | Acc: 81.27% | Loss: 0.6456 | Time: 61.70s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:35:40] [Round 030] FedProx | Acc: 81.88% | Loss: 0.5110 | Time: 56.52s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:36:39] [Round 031] FedProx | Acc: 80.88% | Loss: 0.6331 | Time: 58.70s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:38:13] [Round 032] FedProx | Acc: 87.11% | Loss: 0.4438 | Time: 93.82s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:39:53] [Round 033] FedProx | Acc: 82.23% | Loss: 0.5280 | Time: 100.18s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:40:57] [Round 034] FedProx | Acc: 81.28% | Loss: 0.5859 | Time: 64.20s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:41:49] [Round 035] FedProx | Acc: 82.44% | Loss: 0.4909 | Time: 51.97s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:43:02] [Round 036] FedProx | Acc: 79.46% | Loss: 0.5197 | Time: 73.21s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:44:02] [Round 037] FedProx | Acc: 83.05% | Loss: 0.5080 | Time: 60.14s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:45:02] [Round 038] FedProx | Acc: 79.19% | Loss: 0.5421 | Time: 59.79s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:46:03] [Round 039] FedProx | Acc: 86.62% | Loss: 0.3890 | Time: 60.57s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:46:38] [Round 040] FedProx | Acc: 83.45% | Loss: 0.4996 | Time: 35.01s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:46:56] [Round 041] FedProx | Acc: 81.16% | Loss: 0.4897 | Time: 18.82s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:47:19] [Round 042] FedProx | Acc: 89.73% | Loss: 0.3725 | Time: 22.29s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:47:43] [Round 043] FedProx | Acc: 85.15% | Loss: 0.4280 | Time: 23.94s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:48:08] [Round 044] FedProx | Acc: 88.79% | Loss: 0.3766 | Time: 24.84s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:48:36] [Round 045] FedProx | Acc: 88.14% | Loss: 0.3852 | Time: 28.34s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:49:16] [Round 046] FedProx | Acc: 78.76% | Loss: 0.5881 | Time: 40.37s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:50:13] [Round 047] FedProx | Acc: 81.85% | Loss: 0.4775 | Time: 56.61s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:50:27] [Round 048] FedProx | Acc: 87.46% | Loss: 0.4130 | Time: 14.21s | Max GPU Mem: 1507.58 MB
[2025-11-27 01:50:46] [Round 049] FedProx | Acc: 88.24% | Loss: 0.3684 | Time: 18.70s | Max GPU Mem: 1507.46 MB
[2025-11-27 01:51:12] [Round 050] FedProx | Acc: 86.75% | Loss: 0.3754 | Time: 25.74s | Max GPU Mem: 1507.33 MB
[2025-11-27 01:51:12] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedProx_head_mnist_resnet18_s1.npz
