We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b6e9dcb commit b560fffCopy full SHA for b560fff
beginner_source/introyt/introyt1_tutorial.py
@@ -176,9 +176,9 @@ class LeNet(nn.Module):
176
177
def __init__(self):
178
super(LeNet, self).__init__()
179
- # 1 input image channel (black & white), 6 output channels, 3x3 square convolution
+ # 1 input image channel (black & white), 6 output channels, 5x5 square convolution
180
# kernel
181
- self.conv1 = nn.Conv2d(1, 6, 3)
+ self.conv1 = nn.Conv2d(1, 6, 5)
182
self.conv2 = nn.Conv2d(6, 16, 3)
183
# an affine operation: y = Wx + b
184
self.fc1 = nn.Linear(16 * 6 * 6, 120) # 6*6 from image dimension
0 commit comments