[2025-12-03 03:21:31] Namespace(method='FedNewton', 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/compare_solvers_20251203_001839_resnet18_head/cifar10/logs/diag', optimizer='sgd', hessian_batches=64, damping=0.0001, alpha=0.5, max_norm=0, newton_solver='diag', newton_cg_max_iter=10, lbfgs_m=5)
[2025-12-03 03:21:31] Start: FedNewton | Mode: head | Dataset: cifar10 | Solver: diag
[2025-12-03 03:23:50] [Round 001] FedNewton (Solver: diag) | Acc: 11.99% | Loss: 485.3277 | Time: 25.93s | Max GPU Mem: 1438.79 MB
[2025-12-03 03:23:50] Early stopping as test loss is very high.
[2025-12-03 03:23:50] Saved to runs/compare_solvers_20251203_001839_resnet18_head/cifar10/logs/diag/FedNewton_head_cifar10_resnet18_diag_s2.npz
