fix typo

2017-11-08 22:45:10 +08:00
parent 9318a69f08
commit 2ae0c0fa6a
1 changed files with 1 additions and 1 deletions
--- a/tutorial-contents/401_CNN.py
+++ b/tutorial-contents/401_CNN.py
@ -62,7 +62,7 @@ class CNN(nn.Module):
            nn.ReLU(),                      # activation
            nn.MaxPool2d(kernel_size=2),    # choose max value in 2x2 area, output shape (16, 14, 14)
        )
-        self.conv2 = nn.Sequential(         # input shape (1, 28, 28)
+        self.conv2 = nn.Sequential(         # input shape (1, 14, 14)
            nn.Conv2d(16, 32, 5, 1, 2),     # output shape (32, 14, 14)
            nn.ReLU(),                      # activation
            nn.MaxPool2d(2),                # output shape (32, 7, 7)