Skip to content

Commit 46c516f

Browse files
committed
fix typo
1 parent 968b6e2 commit 46c516f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tutorial-contents/401_CNN.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def __init__(self):
7171
nn.ReLU(), # activation
7272
nn.MaxPool2d(kernel_size=2), # choose max value in 2x2 area, output shape (16, 14, 14)
7373
)
74-
self.conv2 = nn.Sequential( # input shape (1, 14, 14)
74+
self.conv2 = nn.Sequential( # input shape (16, 14, 14)
7575
nn.Conv2d(16, 32, 5, 1, 2), # output shape (32, 14, 14)
7676
nn.ReLU(), # activation
7777
nn.MaxPool2d(2), # output shape (32, 7, 7)

0 commit comments

Comments
 (0)