Skip to content

Commit 70e0f7d

Browse files
committed
Fix size in convnet comment
Reported on theano-users
1 parent 3aa50bf commit 70e0f7d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

code/convolutional_mlp.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -191,8 +191,9 @@ def evaluate_lenet5(learning_rate=0.1, n_epochs=200,
191191
)
192192

193193
# the HiddenLayer being fully-connected, it operates on 2D matrices of
194-
# shape (batch_size,num_pixels) (i.e matrix of rasterized images).
195-
# This will generate a matrix of shape (20,32*4*4) = (20,512)
194+
# shape (batch_size, num_pixels) (i.e matrix of rasterized images).
195+
# This will generate a matrix of shape (batch_size, nkerns[1] * 4 * 4),
196+
# or (500, 50 * 4 * 4) = (500, 800) with the default values.
196197
layer2_input = layer1.output.flatten(2)
197198

198199
# construct a fully-connected sigmoidal layer

0 commit comments

Comments
 (0)