We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 968b6e2 commit 46c516fCopy full SHA for 46c516f
1 file changed
tutorial-contents/401_CNN.py
@@ -71,7 +71,7 @@ def __init__(self):
71
nn.ReLU(), # activation
72
nn.MaxPool2d(kernel_size=2), # choose max value in 2x2 area, output shape (16, 14, 14)
73
)
74
- self.conv2 = nn.Sequential( # input shape (1, 14, 14)
+ self.conv2 = nn.Sequential( # input shape (16, 14, 14)
75
nn.Conv2d(16, 32, 5, 1, 2), # output shape (32, 14, 14)
76
77
nn.MaxPool2d(2), # output shape (32, 7, 7)
0 commit comments