Skip to content

Commit 0c8798c

Browse files
authored
fix NN architecture image vs code mismatch
The network architecture shows images of 5x5 size, while in code, it is written 6x6. Later in the tutorial, the correct architecture of 5x5 is shown.
1 parent b440a38 commit 0c8798c

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
@@ -51,7 +51,7 @@ def __init__(self):
5151
self.conv1 = nn.Conv2d(1, 6, 3)
5252
self.conv2 = nn.Conv2d(6, 16, 3)
5353
# an affine operation: y = Wx + b
54-
self.fc1 = nn.Linear(16 * 6 * 6, 120) # 6*6 from image dimension
54+
self.fc1 = nn.Linear(16 * 5 * 5, 120) # 5*5 from image dimension
5555
self.fc2 = nn.Linear(120, 84)
5656
self.fc3 = nn.Linear(84, 10)
5757

0 commit comments

Comments
 (0)