Skip to content

Commit 70785d0

Browse files
authored
change kernel size to 5x5 in Pruning example (#2639)
1 parent 53e4142 commit 70785d0

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

intermediate_source/pruning_tutorial.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@
4444
class LeNet(nn.Module):
4545
def __init__(self):
4646
super(LeNet, self).__init__()
47-
# 1 input image channel, 6 output channels, 3x3 square conv kernel
48-
self.conv1 = nn.Conv2d(1, 6, 3)
49-
self.conv2 = nn.Conv2d(6, 16, 3)
47+
# 1 input image channel, 6 output channels, 5x5 square conv kernel
48+
self.conv1 = nn.Conv2d(1, 6, 5)
49+
self.conv2 = nn.Conv2d(6, 16, 5)
5050
self.fc1 = nn.Linear(16 * 5 * 5, 120) # 5x5 image dimension
5151
self.fc2 = nn.Linear(120, 84)
5252
self.fc3 = nn.Linear(84, 10)

0 commit comments

Comments
 (0)