From 5c1eff351b035db5881d0cff81b1d9c9e150e2d0 Mon Sep 17 00:00:00 2001 From: Teddy Koker Date: Fri, 16 Oct 2020 14:36:49 -0400 Subject: [PATCH] Add load_from_checkpoint function to Loading docs (#4196) * add autodoc load_from_checkpoint to docs * autofunction -> automethod Co-authored-by: Jeff Yang Co-authored-by: Jeff Yang --- docs/source/weights_loading.rst | 2 ++ 1 file changed, 2 insertions(+) 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 ========================