AsyncCheckpointIO¶
- class lightning.pytorch.plugins.io.AsyncCheckpointIO(checkpoint_io=None)[source]¶
Bases:
lightning.pytorch.plugins.io.wrapper._WrappingCheckpointIO
AsyncCheckpointIO
enables 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.