Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add load_from_checkpoint function to Loading docs (#4196) 

* add autodoc load_from_checkpoint to docs

* autofunction -> automethod

Co-authored-by: Jeff Yang <ydcjeff@outlook.com>

Co-authored-by: Jeff Yang <ydcjeff@outlook.com>
This commit is contained in:
Teddy Koker 2020-10-16 14:36:49 -04:00 committed by GitHub
parent 827a557269
commit 5c1eff351b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/weights_loading.rst
View File

@ -176,6 +176,8 @@ you can restore the model like this
# uses in_dim=128, out_dim=10 # uses in_dim=128, out_dim=10
model = LitModel.load_from_checkpoint(PATH, 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 Restoring Training State
======================== ========================