[2025-11-27 03:22:20] Namespace(method='FedNewton', 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=64, damping=0.0001, alpha=0.1)
[2025-11-27 03:22:20] Start: FedNewton | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 03:24:52] [Round 001] FedNewton | Acc: 91.87% | Loss: 0.4773 | Time: 40.45s | Max GPU Mem: 1463.29 MB
[2025-11-27 03:25:30] [Round 002] FedNewton | Acc: 95.82% | Loss: 0.2460 | Time: 38.01s | Max GPU Mem: 1463.33 MB
[2025-11-27 03:26:26] [Round 003] FedNewton | Acc: 91.37% | Loss: 0.2709 | Time: 55.75s | Max GPU Mem: 1463.37 MB
[2025-11-27 03:27:18] [Round 004] FedNewton | Acc: 94.20% | Loss: 0.1921 | Time: 52.21s | Max GPU Mem: 1463.37 MB
[2025-11-27 03:28:25] [Round 005] FedNewton | Acc: 86.72% | Loss: 0.5837 | Time: 66.75s | Max GPU Mem: 1463.39 MB
[2025-11-27 03:29:12] [Round 006] FedNewton | Acc: 61.39% | Loss: 7.7858 | Time: 47.38s | Max GPU Mem: 1463.39 MB
[2025-11-27 03:29:39] [Round 007] FedNewton | Acc: 32.03% | Loss: 28.0476 | Time: 26.73s | Max GPU Mem: 1463.39 MB
[2025-11-27 03:29:39] Early stopping as test loss is very high.
[2025-11-27 03:29:39] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedNewton_head_mnist_resnet18_s1.npz
