From 47d08825152eddc3077a019830b8dc453a223871 Mon Sep 17 00:00:00 2001 From: Yuxin Wu Date: Wed, 30 Sep 2020 22:00:56 -0700 Subject: [PATCH] Update saving_loading_models.py --- beginner_source/saving_loading_models.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/beginner_source/saving_loading_models.py b/beginner_source/saving_loading_models.py index 74a8edc892d..c94d2d8ce36 100644 --- a/beginner_source/saving_loading_models.py +++ b/beginner_source/saving_loading_models.py @@ -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