Skip to content

Commit 15b0170

Browse files
Fix dimensions of 1st Linear layer to match new expected size
1 parent c5435bf commit 15b0170

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/introyt/introyt1_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def __init__(self):
181181
self.conv1 = nn.Conv2d(1, 6, 5)
182182
self.conv2 = nn.Conv2d(6, 16, 5)
183183
# an affine operation: y = Wx + b
184-
self.fc1 = nn.Linear(16 * 6 * 6, 120) # 6*6 from image dimension
184+
self.fc1 = nn.Linear(16 * 5 * 5, 120) # 5*5 from image dimension
185185
self.fc2 = nn.Linear(120, 84)
186186
self.fc3 = nn.Linear(84, 10)
187187

0 commit comments

Comments
 (0)