small fix: torch.nn.Parameter instead of nn.Linear (#1302)

manju-dev · brianjo · holly1238 · web-flow · commit cecea517b5a0 · 2021-04-07T12:54:41.000-07:00
* nn.Linear module is not defined in Polynomial3 model.
* torch.nn.Parameter is used to defined the members of the model

Co-authored-by: Brian Johnson &lt;brianjo@fb.com&gt;
Co-authored-by: holly1238 &lt;77758406+holly1238@users.noreply.github.com&gt;
diff --git a/beginner_source/examples_nn/polynomial_module.py b/beginner_source/examples_nn/polynomial_module.py
@@ -49,8 +49,8 @@ def string(self):
 model = Polynomial3()
 
 # Construct our loss function and an Optimizer. The call to model.parameters()
-# in the SGD constructor will contain the learnable parameters of the nn.Linear
-# module which is members of the model.
+# in the SGD constructor will contain the learnable parameters (defined 
+# with torch.nn.Parameter) which are members of the model.
 criterion = torch.nn.MSELoss(reduction='sum')
 optimizer = torch.optim.SGD(model.parameters(), lr=1e-6)
 for t in range(2000):
