From f4735ad18f21aca5380f3e67aa7174618225dbfe Mon Sep 17 00:00:00 2001 From: Kevin Lin Date: Tue, 22 Mar 2022 07:41:46 +0800 Subject: [PATCH] Update 17_save_load.py Fix a typo --- 17_save_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/17_save_load.py b/17_save_load.py index 95716fa8..061bef81 100644 --- a/17_save_load.py +++ b/17_save_load.py @@ -81,7 +81,7 @@ def forward(self, x): torch.save(checkpoint, FILE) model = Model(n_input_features=6) -optimizer = optimizer = torch.optim.SGD(model.parameters(), lr=0) +optimizer = torch.optim.SGD(model.parameters(), lr=0) checkpoint = torch.load(FILE) model.load_state_dict(checkpoint['model_state'])