Skip to content

Commit 63585a0

Browse files
Merge pull request patrickloeber#14 from kevinlin19910227/kevinlin19910227-patch-1
Update 17_save_load.py
2 parents cc18036 + f4735ad commit 63585a0

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)