# Fix 2 incorrect argument positioning

davychxn · davychxn · commit 9391f96f31e8 · 2024-09-07T20:20:34.000+08:00
1. "weights_only" is augument of torch.load, not for model.load_state_dict Here is the doc: https://pytorch.org/docs/stable/generated/torch.load.html
diff --git a/beginner_source/saving_loading_models.py b/beginner_source/saving_loading_models.py
@@ -153,7 +153,7 @@
 # .. code:: python
 #
 #    model = TheModelClass(*args, **kwargs)
-#    model.load_state_dict(torch.load(PATH), weights_only=True)
+#    model.load_state_dict(torch.load(PATH, weights_only=True))
 #    model.eval()
 #
 # .. note::
@@ -407,7 +407,7 @@
 # .. code:: python
 #
 #    modelB = TheModelBClass(*args, **kwargs)
-#    modelB.load_state_dict(torch.load(PATH), strict=False, weights_only=True)
+#    modelB.load_state_dict(torch.load(PATH, weights_only=True), strict=False)
 #
 # Partially loading a model or loading a partial model are common
 # scenarios when transfer learning or training a new complex model.

Original file line number	Diff line number	Diff line change
`@@ -153,7 +153,7 @@`
`153`	`153`	`# .. code:: python`
`154`	`154`	`#`
`155`	`155`	`# 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))`
`157`	`157`	`# model.eval()`
`158`	`158`	`#`
`159`	`159`	`# .. note::`
`@@ -407,7 +407,7 @@`
`407`	`407`	`# .. code:: python`
`408`	`408`	`#`
`409`	`409`	`# modelB = TheModelBClass(args, *kwargs)`
`410`		`-# modelB.load_state_dict(torch.load(PATH), strict=False, weights_only=True)`
	`410`	`+# modelB.load_state_dict(torch.load(PATH, weights_only=True), strict=False)`
`411`	`411`	`#`
`412`	`412`	`# Partially loading a model or loading a partial model are common`
`413`	`413`	`# scenarios when transfer learning or training a new complex model.`