Fix training resuming docs (#1265)
This commit is contained in:
parent
fb42872259
commit
b74a3c5106
|
@ -84,9 +84,7 @@ To save your own checkpoint call:
|
||||||
Checkpoint Loading
|
Checkpoint Loading
|
||||||
------------------
|
------------------
|
||||||
|
|
||||||
You might want to not only load a model but also continue training it. Use this method to
|
To load a model along with its weights, biases and hyperparameters use following method:
|
||||||
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).
|
|
||||||
|
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
|
@ -96,3 +94,7 @@ However, the dataloaders will start from the first batch again (if you shuffled
|
||||||
|
|
||||||
A LightningModule is no different than a nn.Module. This means you can load it and use it for
|
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`.
|
|
@ -39,15 +39,9 @@ Lightning will restore the session if you pass a logger with the same version an
|
||||||
.. code-block:: python
|
.. code-block:: python
|
||||||
|
|
||||||
from pytorch_lightning import Trainer
|
from pytorch_lightning import Trainer
|
||||||
from pytorch_lightning.loggers import TestTubeLogger
|
|
||||||
|
|
||||||
logger = TestTubeLogger(
|
|
||||||
save_dir='./savepath',
|
|
||||||
version=1 # An existing version with a saved checkpoint
|
|
||||||
)
|
|
||||||
trainer = Trainer(
|
trainer = Trainer(
|
||||||
logger=logger,
|
resume_from_checkpoint=PATH
|
||||||
default_save_path='./savepath'
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# this fit call loads model weights and trainer state
|
# this fit call loads model weights and trainer state
|
||||||
|
|
Loading…
Reference in New Issue