safe_torch_load

composer.utils.safe_torch_load(composer_states_filepath, map_location='cpu', load_fsdp_monolith_rank0_only=False)

Load a torch checkpoint, catching errors due to backwards compatibility issues.

Parameters

• composer_states_filepath – The path to the checkpoint file.

• map_location – The location to load the checkpoint to.

• load_fsdp_monolith_rank0_only – Whether the checkpoint is a monolith FSDP checkpoint.