Table of Contents
2.5.1.post0

Home

  • Welcome to ⚡ Lightning Fabric
  • Install

Get started in steps

  • Basic skills
  • Intermediate skills
  • Advanced skills

Core API Reference

  • Fabric Arguments
  • Fabric Methods

Full API Reference

  • Accelerators
  • Collectives
  • Environments
  • Fabric
  • IO
  • Loggers
  • Precision
  • Strategies
  • Utilities

More

  • Examples
  • Glossary
  • How-tos
  • Style Guide
  • Overview
  • Team management
  • Production
  • Security
  • Open source
    • Overview
    • PyTorch Lightning
    • Fabric
    • Lit-GPT
    • Torchmetrics
    • Litdata
    • Lit LLaMA
    • Litserve
  • Examples
  • Glossary
  • FAQ
  • Docs >
  • lightning.fabric.plugins.io
Shortcuts

lightning.fabric.plugins.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.

  • lightning.fabric.plugins.io
    • IO

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.