megatron.checkpointing.save_checkpoint#

megatron.checkpointing.save_checkpoint(iteration, model, optimizer, opt_param_scheduler)#

Save a model checkpoint.