Skip to content

Commit 6fa035e

Browse files
authored
Update modelsyt_tutorial.py (#1796)
Fixes #1771.
1 parent 86abcf4 commit 6fa035e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/introyt/modelsyt_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ class LeNet(torch.nn.Module):
151151

152152
def __init__(self):
153153
super(LeNet, self).__init__()
154-
# 1 input image channel (black & white), 6 output channels, 3x3 square convolution
154+
# 1 input image channel (black & white), 6 output channels, 5x5 square convolution
155155
# kernel
156156
self.conv1 = torch.nn.Conv2d(1, 6, 5)
157157
self.conv2 = torch.nn.Conv2d(6, 16, 3)

0 commit comments

Comments
 (0)