Skip to content

Commit 7f49279

Browse files
authored
the output of s2 should not change the number of feature maps (#2855)
`c1` outputs a Tensor with size (N, 6, 28, 28), and `s2` should not change the 6 into 16.
1 parent d2d26f5 commit 7f49279

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/blitz/neural_networks_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def forward(self, input):
6161
# outputs a Tensor with size (N, 6, 28, 28), where N is the size of the batch
6262
c1 = F.relu(self.conv1(input))
6363
# Subsampling layer S2: 2x2 grid, purely functional,
64-
# this layer does not have any parameter, and outputs a (N, 16, 14, 14) Tensor
64+
# this layer does not have any parameter, and outputs a (N, 6, 14, 14) Tensor
6565
s2 = F.max_pool2d(c1, (2, 2))
6666
# Convolution layer C3: 6 input channels, 16 output channels,
6767
# 5x5 square convolution, it uses RELU activation function, and

0 commit comments

Comments
 (0)