Skip to content

Commit c244ef9

Browse files
Sachapernaksvekars
andauthored
Fix incorrect code comment in neural_networks_tutorial (#3679)
The code example incorrectly labels the final linear layer as a "Gaussian layer". However, `fc3` is a standard fully connected layer (`nn.Linear(84, 10)`). Updated the comment to "Fully connected layer" to avoid confusion. cc @albanD @jbschlosser Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
1 parent 4fa1fa8 commit c244ef9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/blitz/neural_networks_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def forward(self, input):
7878
# Fully connected layer F6: (N, 120) Tensor input,
7979
# and outputs a (N, 84) Tensor, it uses RELU activation function
8080
f6 = F.relu(self.fc2(f5))
81-
# Gaussian layer OUTPUT: (N, 84) Tensor input, and
81+
# Fully connected layer OUTPUT: (N, 84) Tensor input, and
8282
# outputs a (N, 10) Tensor
8383
output = self.fc3(f6)
8484
return output

0 commit comments

Comments
 (0)