Skip to content

Commit e56e6a9

Browse files
committed
Fix arguments in 2d CNN
1 parent a37a0c6 commit e56e6a9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

example/cnn_mnist.f90

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,9 @@ program cnn_mnist
2222
input(784), &
2323
reshape(1, 28, 28), &
2424
conv(filters=8, kernel_width=3, kernel_height=3, activation=relu()), &
25-
maxpool(pool_width=2, stride=2), &
25+
maxpool(pool_width=2, pool_height=2, stride=2), &
2626
conv(filters=16, kernel_width=3, kernel_height=3, activation=relu()), &
27-
maxpool(pool_width=2, stride=2), &
27+
maxpool(pool_width=2, pool_height=2, stride=2), &
2828
dense(10, activation=softmax()) &
2929
])
3030

0 commit comments

Comments
 (0)