megatron.checkpointing.check_checkpoint_args#

megatron.checkpointing.check_checkpoint_args(checkpoint_args)#

Ensure fixed arguments for a model are the same for the input arguments and the one retrieved from checkpoint.