Skip to content

Commit d549c38

Browse files
fzyzcjySvetlana Karslioglu
and
Svetlana Karslioglu
authored
Update buildmodel_tutorial.py (#2131)
Co-authored-by: Svetlana Karslioglu <svekars@fb.com>
1 parent f26db6b commit d549c38

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/basics/buildmodel_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ def forward(self, x):
7777
# along with some `background operations <https://github.com/pytorch/pytorch/blob/270111b7b611d174967ed204776985cefca9c144/torch/nn/modules/module.py#L866>`_.
7878
# Do not call ``model.forward()`` directly!
7979
#
80-
# Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to each output of 10 raw predicted values for each class, and dim=1 corresponding to the individual values of each output. .
80+
# Calling the model on the input returns a 2-dimensional tensor with dim=0 corresponding to each output of 10 raw predicted values for each class, and dim=1 corresponding to the individual values of each output.
8181
# We get the prediction probabilities by passing it through an instance of the ``nn.Softmax`` module.
8282

8383
X = torch.rand(1, 28, 28, device=device)

0 commit comments

Comments
 (0)