[2025-12-03 03:20:36] Namespace(method='FedNewton', training_mode='head', dataset='cifar10', 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/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:20:36] Start: FedNewton | Mode: head | Dataset: cifar10 | Solver: diag
[2025-12-03 03:23:13] [Round 001] FedNewton (Solver: diag) | Acc: 14.78% | Loss: 152.0592 | Time: 43.18s | Max GPU Mem: 1438.79 MB
[2025-12-03 03:23:13] Early stopping as test loss is very high.
[2025-12-03 03:23:13] Saved to runs/compare_solvers_20251203_001839_resnet18_head/cifar10/logs/diag/FedNewton_head_cifar10_resnet18_diag_s1.npz
