AsyncCheckpointIO
- class lightning.pytorch.plugins.io.AsyncCheckpointIO(checkpoint_io=None)
Bases:
_WrappingCheckpointIOAsyncCheckpointIOenables saving the checkpoints asynchronously in a thread.Warning
This is an experimental feature.
- Parameters:
checkpoint_io (
Optional[CheckpointIO]) – A checkpoint IO plugin that is used as the basis for async checkpointing.
- save_checkpoint(*args, **kwargs)
Uses the
ThreadPoolExecutorto save the checkpoints using the basecheckpoint_io.- Return type:
- teardown()
This method is called to close the threads.
- Return type:
Was this page helpful?
Thanks for the feedback.