[2025-11-27 03:23:46] Namespace(method='FedNewton', training_mode='head', dataset='mnist', 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_001404_resnet18_head/mnist/logs', optimizer='sgd', hessian_batches=64, damping=0.0001, alpha=0.1)
[2025-11-27 03:23:46] Start: FedNewton | Mode: head | Dataset: mnist | Device: cuda
[2025-11-27 03:26:36] [Round 001] FedNewton | Acc: 60.14% | Loss: 1.2217 | Time: 57.37s | Max GPU Mem: 1463.29 MB
[2025-11-27 03:27:35] [Round 002] FedNewton | Acc: 81.27% | Loss: 1.1569 | Time: 59.75s | Max GPU Mem: 1463.33 MB
[2025-11-27 03:28:27] [Round 003] FedNewton | Acc: 82.97% | Loss: 0.7248 | Time: 51.44s | Max GPU Mem: 1463.33 MB
[2025-11-27 03:29:13] [Round 004] FedNewton | Acc: 50.44% | Loss: 11.1724 | Time: 46.54s | Max GPU Mem: 1463.37 MB
[2025-11-27 03:29:13] Early stopping as test loss is very high.
[2025-11-27 03:29:13] Saved to runs/run_20251127_001404_resnet18_head/mnist/logs/FedNewton_head_mnist_resnet18_s2.npz
