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.
Bases: TorchCheckpointIO
CheckpointIO that utilizes xm.save
to save checkpoints for TPU training strategies.
Warning
This is an experimental feature.