XLACheckpointIO¶
- class lightning.pytorch.plugins.io.XLACheckpointIO(*args, **kwargs)[source]¶
Bases:
lightning.fabric.plugins.io.torch_io.TorchCheckpointIO
CheckpointIO that utilizes
xm.save()
to save checkpoints for TPU training strategies.Warning
This is an experimental feature.