We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 53e4142 commit 70785d0Copy full SHA for 70785d0
intermediate_source/pruning_tutorial.py
@@ -44,9 +44,9 @@
44
class LeNet(nn.Module):
45
def __init__(self):
46
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)
+ # 1 input image channel, 6 output channels, 5x5 square conv kernel
+ self.conv1 = nn.Conv2d(1, 6, 5)
+ self.conv2 = nn.Conv2d(6, 16, 5)
50
self.fc1 = nn.Linear(16 * 5 * 5, 120) # 5x5 image dimension
51
self.fc2 = nn.Linear(120, 84)
52
self.fc3 = nn.Linear(84, 10)
0 commit comments