Skip to content

Commit f4735ad

Browse files
Update 17_save_load.py
Fix a typo
1 parent e6dd99c commit f4735ad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

17_save_load.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def forward(self, x):
8181
torch.save(checkpoint, FILE)
8282

8383
model = Model(n_input_features=6)
84-
optimizer = optimizer = torch.optim.SGD(model.parameters(), lr=0)
84+
optimizer = torch.optim.SGD(model.parameters(), lr=0)
8585

8686
checkpoint = torch.load(FILE)
8787
model.load_state_dict(checkpoint['model_state'])

0 commit comments

Comments
 (0)