megatron.checkpointing.get_checkpoint_names

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

Determine the directory name for this rank’s checkpoint.