save checkpoints while training hyperparams