- class lightning.fabric.plugins.io.checkpoint_io.CheckpointIO¶
Interface to save/load checkpoints as they are saved through the
This is an experimental feature.
Typically most plugins either use the Torch based IO Plugin;
TorchCheckpointIObut may require particular handling depending on the plugin.
In addition, you can pass a custom
CheckpointIOby extending this class and passing it to the Trainer, i.e
For some plugins, it is not possible to use a custom checkpoint plugin as checkpointing logic is not modifiable.
- abstract load_checkpoint(path, map_location=None)¶
Load checkpoint from a path when resuming or loading ckpt for test/validate/predict stages.
- Return type:
Returns: The loaded checkpoint.
- abstract remove_checkpoint(path)¶
Remove checkpoint file from the filesystem.
- abstract save_checkpoint(checkpoint, path, storage_options=None)¶
Save model/training states as a checkpoint file through state-dump and file-write.