XLACheckpointIO¶
- class pytorch_lightning.plugins.io.XLACheckpointIO[source]¶
Bases:
pytorch_lightning.plugins.io.torch_plugin.TorchCheckpointIO
CheckpointIO that utilizes
xm.save()
to save checkpoints for TPU training strategies.
Bases: pytorch_lightning.plugins.io.torch_plugin.TorchCheckpointIO
CheckpointIO that utilizes xm.save()
to save checkpoints for TPU training strategies.