Merge branch 'main' into issue2637

pytorch · Nov 2, 2023 · 6ac6c37 · 6ac6c37
2 parents d9c439b + 70785d0
commit 6ac6c37
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/intermediate_source/pruning_tutorial.py b/intermediate_source/pruning_tutorial.py
@@ -44,9 +44,9 @@
 class LeNet(nn.Module):
     def __init__(self):
         super(LeNet, self).__init__()
-        # 1 input image channel, 6 output channels, 3x3 square conv kernel
-        self.conv1 = nn.Conv2d(1, 6, 3)
-        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)
         self.fc1 = nn.Linear(16 * 5 * 5, 120)  # 5x5 image dimension
         self.fc2 = nn.Linear(120, 84)
         self.fc3 = nn.Linear(84, 10)