diff --git a/docs/source/weights_loading.rst b/docs/source/weights_loading.rst index f6ba221bee..8f44c10853 100644 --- a/docs/source/weights_loading.rst +++ b/docs/source/weights_loading.rst @@ -176,6 +176,8 @@ you can restore the model like this # uses in_dim=128, out_dim=10 model = LitModel.load_from_checkpoint(PATH, in_dim=128, out_dim=10) +.. automethod:: pytorch_lightning.core.lightning.LightningModule.load_from_checkpoint + :noindex: Restoring Training State ========================