TorchCheckpointIO¶
- class pytorch_lightning.plugins.io.TorchCheckpointIO[source]¶
Bases:
lightning_lite.plugins.io.checkpoint_io.CheckpointIO
CheckpointIO that utilizes
torch.save()
andtorch.load()
to save and load checkpoints respectively, common for most use cases.- load_checkpoint(path, map_location=<function TorchCheckpointIO.<lambda>>)[source]¶
Loads checkpoint using
torch.load()
, with additional handling forfsspec
remote loading of files.- Parameters:
- Return type:
Returns: The loaded checkpoint.
- Raises:
FileNotFoundError – If
path
is not found by thefsspec
filesystem