Change variable name from net to model (#2243)

kiersten-stokes · Svetlana Karslioglu · web-flow · commit 8475fd934610 · 2023-03-29T09:27:12.000-07:00
Co-authored-by: Svetlana Karslioglu &lt;svekars@fb.com&gt;
diff --git a/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py b/recipes_source/recipes/saving_and_loading_a_general_checkpoint.py
@@ -129,7 +129,7 @@ def forward(self, x):
 # 
 
 model = Net()
-optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)
+optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9)
 
 checkpoint = torch.load(PATH)
 model.load_state_dict(checkpoint['model_state_dict'])

Original file line number	Diff line number	Diff line change
`@@ -129,7 +129,7 @@ def forward(self, x):`
`129`	`129`	`#`
`130`	`130`
`131`	`131`	`model = Net()`
`132`		`-optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)`
	`132`	`+optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9)`
`133`	`133`
`134`	`134`	`checkpoint = torch.load(PATH)`
`135`	`135`	`model.load_state_dict(checkpoint['model_state_dict'])`