Skip to content

Commit 59209c6

Browse files
authored
Change kernel filters to reflect image and Colab
The kernel filters should be 5x5 as opposed to 3x3 as that is what is depicted in the image and what it done within the Google Colab.
1 parent 99d15ce commit 59209c6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

beginner_source/blitz/neural_networks_tutorial.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,10 @@ class Net(nn.Module):
4646

4747
def __init__(self):
4848
super(Net, self).__init__()
49-
# 1 input image channel, 6 output channels, 3x3 square convolution
49+
# 1 input image channel, 6 output channels, 5x5 square convolution
5050
# kernel
51-
self.conv1 = nn.Conv2d(1, 6, 3)
52-
self.conv2 = nn.Conv2d(6, 16, 3)
51+
self.conv1 = nn.Conv2d(1, 6, 5)
52+
self.conv2 = nn.Conv2d(6, 16, 5)
5353
# an affine operation: y = Wx + b
5454
self.fc1 = nn.Linear(16 * 6 * 6, 120) # 6*6 from image dimension
5555
self.fc2 = nn.Linear(120, 84)

0 commit comments

Comments
 (0)