diff --git a/docs/source/extensions/plugins.rst b/docs/source/extensions/plugins.rst index 252fb47570fc4..d81ed52fae504 100644 --- a/docs/source/extensions/plugins.rst +++ b/docs/source/extensions/plugins.rst @@ -74,6 +74,20 @@ Precision Plugins IPUPrecisionPlugin +CheckpointIO Plugins +-------------------- + +.. currentmodule:: pytorch_lightning.plugins.io + +.. autosummary:: + :nosignatures: + :template: classtemplate.rst + + CheckpointIO + TorchCheckpointIO + XLACheckpointIO + + Cluster Environments --------------------