[2025-11-27 03:21:41] Namespace(method='FedNewton', training_mode='head', dataset='mnist', 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/mnist/logs', optimizer='sgd', hessian_batches=64, damping=0.0001, alpha=0.1)
[2025-11-27 03:21:41] Start: FedNewton | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 03:24:03] [Round 001] FedNewton | Acc: 58.46% | Loss: 1.3926 | Time: 30.57s | Max GPU Mem: 1463.29 MB
[2025-11-27 03:24:33] [Round 002] FedNewton | Acc: 70.79% | Loss: 1.7696 | Time: 30.33s | Max GPU Mem: 1463.33 MB
[2025-11-27 03:25:15] [Round 003] FedNewton | Acc: 77.20% | Loss: 1.1454 | Time: 42.10s | Max GPU Mem: 1463.35 MB
[2025-11-27 03:26:03] [Round 004] FedNewton | Acc: 75.79% | Loss: 0.9947 | Time: 48.08s | Max GPU Mem: 1463.37 MB
[2025-11-27 03:26:52] [Round 005] FedNewton | Acc: 63.66% | Loss: 4.3286 | Time: 48.74s | Max GPU Mem: 1463.37 MB
[2025-11-27 03:28:11] [Round 006] FedNewton | Acc: 23.36% | Loss: 36.9459 | Time: 78.59s | Max GPU Mem: 1463.39 MB
[2025-11-27 03:28:11] Early stopping as test loss is very high.
[2025-11-27 03:28:11] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedNewton_head_mnist_resnet18_s0.npz
