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

Fix training resuming docs (#1265)

Browse Source
This commit is contained in:
Paweł Rzepiński
2020-03-29 15:29:48 -04:00
committed by GitHub
parent fb42872259
commit b74a3c5106
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/source/weights_loading.rst
+6 -4
View File
@@ -84,9 +84,7 @@ To save your own checkpoint call:
Checkpoint Loading
------------------
You might want to not only load a model but also continue training it. Use this method to
restore the trainer state as well. This will continue from the epoch and global step you last left off.
However, the dataloaders will start from the first batch again (if you shuffled it shouldn't matter).
To load a model along with its weights, biases and hyperparameters use following method:
.. code-block:: python
@@ -95,4 +93,8 @@ However, the dataloaders will start from the first batch again (if you shuffled
y_hat = model(x)
A LightningModule is no different than a nn.Module. This means you can load it and use it for
predictions as you would a nn.Module.
predictions as you would a nn.Module.
.. note:: To restore the trainer state as well use
:meth:`pytorch_lightning.trainer.trainer.Trainer.resume_from_checkpoint`.