safe_torch_load#

composer.utils.safe_torch_load(composer_states_filepath, map_location='cpu')[source]#

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.