wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix docs on saving checkpoints manually (#1373)

Browse Source
This commit is contained in:
Jeremy Jordan
2020-04-05 11:10:44 -04:00
committed by GitHub
parent b18accc64c
commit 38c56081ac
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/weights_loading.rst
+5 -3
View File
@@ -74,12 +74,14 @@ The Lightning checkpoint also saves the hparams (hyperparams) passed into the Li
Manual saving
^^^^^^^^^^^^^
To save your own checkpoint call:
You can manually save checkpoints and restore your model from the checkpointed state.
.. code-block:: python
model.save_checkpoint(PATH)
model = MyModel(hparams)
trainer.fit(model)
trainer.save_checkpoint("example.ckpt")
new_model = MyModel.load_from_checkpoint(checkpoint_path="example.ckpt")
Checkpoint Loading
------------------