XLACheckpointIO
- class lightning.pytorch.plugins.io.XLACheckpointIO(*args, **kwargs)[source]
Bases:
TorchCheckpointIO
CheckpointIO that utilizes
xm.save
to save checkpoints for TPU training strategies.Warning
This is an experimental feature.
- save_checkpoint(checkpoint, path, storage_options=None)[source]
Save model/training states as a checkpoint file through state-dump and file-write.