megatron.checkpointing.get_checkpoint_name

megatron.checkpointing.get_checkpoint_name(checkpoints_path, iteration, release=False, pipeline_parallel=None, tensor_rank=None, pipeline_rank=None)

Determine the directory name for this rank’s checkpoint.