[2025-11-27 09:16:59] 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_001608_resnet18_head/cifar10/logs', optimizer='sgd', hessian_batches=32, damping=0.01, alpha=0.5)
[2025-11-27 09:16:59] Start: FedProx | Mode: head | Dataset: cifar10 | Device: cuda
[2025-11-27 09:19:39] [Round 001] FedProx | Acc: 51.58% | Loss: 1.4448 | Time: 46.88s | Max GPU Mem: 1482.23 MB
[2025-11-27 09:20:22] [Round 002] FedProx | Acc: 63.64% | Loss: 1.1776 | Time: 42.71s | Max GPU Mem: 1482.27 MB
[2025-11-27 09:21:35] [Round 003] FedProx | Acc: 74.85% | Loss: 0.8892 | Time: 73.01s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:22:39] [Round 004] FedProx | Acc: 72.94% | Loss: 0.8550 | Time: 63.81s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:23:56] [Round 005] FedProx | Acc: 78.18% | Loss: 0.7148 | Time: 76.92s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:24:51] [Round 006] FedProx | Acc: 77.13% | Loss: 0.7066 | Time: 54.99s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:25:55] [Round 007] FedProx | Acc: 78.83% | Loss: 0.6612 | Time: 64.61s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:26:40] [Round 008] FedProx | Acc: 80.19% | Loss: 0.6211 | Time: 45.26s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:27:29] [Round 009] FedProx | Acc: 78.04% | Loss: 0.6507 | Time: 48.20s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:28:26] [Round 010] FedProx | Acc: 79.58% | Loss: 0.6107 | Time: 57.61s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:29:42] [Round 011] FedProx | Acc: 79.41% | Loss: 0.6126 | Time: 75.56s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:30:29] [Round 012] FedProx | Acc: 80.75% | Loss: 0.5786 | Time: 46.96s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:31:31] [Round 013] FedProx | Acc: 79.89% | Loss: 0.5888 | Time: 62.34s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:32:20] [Round 014] FedProx | Acc: 77.66% | Loss: 0.6409 | Time: 48.65s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:33:11] [Round 015] FedProx | Acc: 80.94% | Loss: 0.5584 | Time: 50.97s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:34:04] [Round 016] FedProx | Acc: 81.14% | Loss: 0.5533 | Time: 53.11s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:34:53] [Round 017] FedProx | Acc: 81.59% | Loss: 0.5471 | Time: 49.06s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:35:50] [Round 018] FedProx | Acc: 80.94% | Loss: 0.5578 | Time: 56.98s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:36:53] [Round 019] FedProx | Acc: 82.01% | Loss: 0.5326 | Time: 63.55s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:37:53] [Round 020] FedProx | Acc: 82.52% | Loss: 0.5158 | Time: 59.68s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:39:00] [Round 021] FedProx | Acc: 81.41% | Loss: 0.5482 | Time: 66.75s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:40:09] [Round 022] FedProx | Acc: 81.59% | Loss: 0.5318 | Time: 69.16s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:41:18] [Round 023] FedProx | Acc: 81.59% | Loss: 0.5376 | Time: 68.54s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:42:25] [Round 024] FedProx | Acc: 80.80% | Loss: 0.5521 | Time: 67.86s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:43:39] [Round 025] FedProx | Acc: 82.96% | Loss: 0.4998 | Time: 73.60s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:44:32] [Round 026] FedProx | Acc: 81.65% | Loss: 0.5319 | Time: 53.33s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:45:43] [Round 027] FedProx | Acc: 82.54% | Loss: 0.5066 | Time: 71.08s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:46:57] [Round 028] FedProx | Acc: 83.26% | Loss: 0.4880 | Time: 73.48s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:48:06] [Round 029] FedProx | Acc: 83.72% | Loss: 0.4790 | Time: 69.32s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:49:04] [Round 030] FedProx | Acc: 83.02% | Loss: 0.4990 | Time: 57.80s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:50:19] [Round 031] FedProx | Acc: 83.28% | Loss: 0.4859 | Time: 74.57s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:51:36] [Round 032] FedProx | Acc: 84.36% | Loss: 0.4661 | Time: 77.92s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:52:48] [Round 033] FedProx | Acc: 83.59% | Loss: 0.4778 | Time: 71.99s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:53:53] [Round 034] FedProx | Acc: 84.18% | Loss: 0.4672 | Time: 64.53s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:55:06] [Round 035] FedProx | Acc: 83.22% | Loss: 0.4833 | Time: 72.60s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:56:18] [Round 036] FedProx | Acc: 83.58% | Loss: 0.4797 | Time: 72.32s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:57:30] [Round 037] FedProx | Acc: 83.49% | Loss: 0.4813 | Time: 72.54s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:58:21] [Round 038] FedProx | Acc: 83.92% | Loss: 0.4696 | Time: 50.61s | Max GPU Mem: 1482.33 MB
[2025-11-27 09:59:26] [Round 039] FedProx | Acc: 84.06% | Loss: 0.4634 | Time: 64.93s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:00:47] [Round 040] FedProx | Acc: 83.58% | Loss: 0.4758 | Time: 80.66s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:01:51] [Round 041] FedProx | Acc: 84.44% | Loss: 0.4588 | Time: 64.65s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:02:22] [Round 042] FedProx | Acc: 83.68% | Loss: 0.4705 | Time: 30.99s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:02:48] [Round 043] FedProx | Acc: 82.68% | Loss: 0.4964 | Time: 25.63s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:03:12] [Round 044] FedProx | Acc: 83.02% | Loss: 0.4870 | Time: 24.43s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:03:31] [Round 045] FedProx | Acc: 83.94% | Loss: 0.4584 | Time: 18.32s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:03:48] [Round 046] FedProx | Acc: 83.50% | Loss: 0.4669 | Time: 17.22s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:04:05] [Round 047] FedProx | Acc: 82.20% | Loss: 0.5028 | Time: 16.70s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:04:28] [Round 048] FedProx | Acc: 83.45% | Loss: 0.4794 | Time: 23.08s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:04:52] [Round 049] FedProx | Acc: 84.53% | Loss: 0.4533 | Time: 24.33s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:05:17] [Round 050] FedProx | Acc: 83.65% | Loss: 0.4736 | Time: 24.79s | Max GPU Mem: 1482.33 MB
[2025-11-27 10:05:17] Saved to runs/run_20251127_001608_resnet18_head/cifar10/logs/FedProx_head_cifar10_resnet18_s2.npz
