Skip to content

Commit c1756e8

Browse files
committed
Fix error in comment.
1 parent b7481db commit c1756e8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/convolutional_mlp.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def evaluate_lenet5(learning_rate=0.1, n_epochs=200,
179179
# Construct the second convolutional pooling layer
180180
# filtering reduces the image size to (12-5+1, 12-5+1) = (8, 8)
181181
# maxpooling reduces this further to (8/2, 8/2) = (4, 4)
182-
# 4D output tensor is thus of shape (nkerns[0], nkerns[1], 4, 4)
182+
# 4D output tensor is thus of shape (batch_size, nkerns[1], 4, 4)
183183
layer1 = LeNetConvPoolLayer(
184184
rng,
185185
input=layer0.output,

0 commit comments

Comments
 (0)