megatron.training.save_checkpoint_and_time# megatron.training.save_checkpoint_and_time(iteration, model, optimizer, opt_param_scheduler)#