AsyncCheckpointIO

class lightning.pytorch.plugins.io.AsyncCheckpointIO(checkpoint_io=None)

기반 클래스: lightning.pytorch.plugins.io.wrapper._WrappingCheckpointIO

AsyncCheckpointIO enables saving the checkpoints asynchronously in a thread.

경고

This is an experimental feature.

매개변수

checkpoint_io (Optional[CheckpointIO]) – A checkpoint IO plugin that is used as the basis for async checkpointing.

save_checkpoint(*args, **kwargs)

Uses the ThreadPoolExecutor to save the checkpoints using the base checkpoint_io.

반환 형식

None

teardown()

This method is called to close the threads.

반환 형식

None