[2025-11-30 01:42:42] Namespace(method='FedNewton', training_mode='head', dataset='cifar10', model_name='tv_resnet50', 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_20251129_235013_tv_resnet50_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-11-30 01:42:42] Start: FedNewton | Mode: head | Dataset: cifar10 | Solver: diag
[2025-11-30 01:45:08] [Round 001] FedNewton (Solver: diag) | Acc: 18.38% | Loss: 1629.2276 | Time: 31.32s | Max GPU Mem: 2564.27 MB
[2025-11-30 01:45:08] Early stopping as test loss is very high.
[2025-11-30 01:45:08] Saved to runs/compare_solvers_20251129_235013_tv_resnet50_head/cifar10/logs/diag/FedNewton_head_cifar10_tv_resnet50_diag_s2.npz
