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.