lightning.fabric.plugins.io

Warning

This is an experimental feature.

IO

CheckpointIO	Interface to save/load checkpoints as they are saved through the Strategy.
TorchCheckpointIO	CheckpointIO that utilizes torch.save() and torch.load() to save and load checkpoints respectively, common for most use cases.
XLACheckpointIO	CheckpointIO that utilizes xm.save to save checkpoints for TPU training strategies.