We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent eaed12b commit 99b2904Copy full SHA for 99b2904
beginner_source/examples_nn/polynomial_module.py
@@ -49,8 +49,8 @@ def string(self):
49
model = Polynomial3()
50
51
# Construct our loss function and an Optimizer. The call to model.parameters()
52
-# in the SGD constructor will contain the learnable parameters of the two
53
-# nn.Linear modules which are members of the model.
+# in the SGD constructor will contain the learnable parameters of the nn.Linear
+# module which is members of the model.
54
criterion = torch.nn.MSELoss(reduction='sum')
55
optimizer = torch.optim.SGD(model.parameters(), lr=1e-6)
56
for t in range(2000):
0 commit comments