fix docs on saving checkpoints manually (#1373)

Lightning-AI · Apr 5, 2020 · 38c5608 · 38c5608
1 parent b18accc
commit 38c5608
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/docs/source/weights_loading.rst b/docs/source/weights_loading.rst
@@ -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
 ------------------