From 2285b35de5f2c3c25ed07d023cdd3306e6945425 Mon Sep 17 00:00:00 2001 From: Kiersten Stokes Date: Tue, 7 Mar 2023 15:19:42 -0600 Subject: [PATCH] Change variable name from net to model --- .../recipes/saving_and_loading_a_general_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py b/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py index a31f43970f6..cc872ae8042 100644 --- a/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py +++ b/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py @@ -129,7 +129,7 @@ def forward(self, x): # model = Net() -optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9) +optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9) checkpoint = torch.load(PATH) model.load_state_dict(checkpoint['model_state_dict'])