Skip to content

Commit

Permalink
Update saving_loading_models.py (pytorch#1173)
Browse files Browse the repository at this point in the history
  • Loading branch information
ppwwyyxx authored Oct 2, 2020
1 parent 3423f67 commit 50555db
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion beginner_source/saving_loading_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,8 @@
# as this contains buffers and parameters that are updated as the model
# trains. Other items that you may want to save are the epoch you left off
# on, the latest recorded training loss, external ``torch.nn.Embedding``
# layers, etc.
# layers, etc. As a result, such a checkpoint is often 2~3 times larger
# than the model alone.
#
# To save multiple components, organize them in a dictionary and use
# ``torch.save()`` to serialize the dictionary. A common PyTorch
Expand Down

0 comments on commit 50555db

Please sign in to comment.