From 46c516fc109809f4299e6bbcd045f8637aef4dd6 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Thu, 8 Mar 2018 00:22:06 +1100 Subject: [PATCH] fix typo --- tutorial-contents/401_CNN.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorial-contents/401_CNN.py b/tutorial-contents/401_CNN.py index 77b1693..6680425 100644 --- a/tutorial-contents/401_CNN.py +++ b/tutorial-contents/401_CNN.py @@ -71,7 +71,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, 14, 14) + self.conv2 = nn.Sequential( # input shape (16, 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)