Skip to content

Commit 7377530

Browse files
berinanieshsvekars
andauthored
fix typo in loading model from disk code (#3021)
Co-authored-by: Svetlana Karslioglu <[email protected]>
1 parent 23a9aa0 commit 7377530

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/saving_loading_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -153,7 +153,7 @@
153153
# .. code:: python
154154
#
155155
# model = TheModelClass(*args, **kwargs)
156-
# model.load_state_dict(torch.load(PATH), weights_only=True)
156+
# model.load_state_dict(torch.load(PATH, weights_only=True))
157157
# model.eval()
158158
#
159159
# .. note::

0 commit comments

Comments
 (0)