Skip to content

Commit d0c7317

Browse files
committed
small fix.
1 parent 307ec52 commit d0c7317

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

code/convolutional_mlp.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ def shared_dataset(data_xy):
224224
# Construct the first convolutional pooling layer:
225225
# filtering reduces the image size to (28-5+1,28-5+1)=(24,24)
226226
# maxpooling reduces this further to (24/2,24/2) = (12,12)
227-
# 4D output tensor is thus of shape (20,20,12,12)
227+
# 4D output tensor is thus of shape (batch_size,20,12,12)
228228
layer0 = LeNetConvPoolLayer(rng, input=layer0_input,
229229
image_shape=(batch_size,1,28,28),
230230
filter_shape=(20,1,5,5), poolsize=(2,2))
@@ -326,7 +326,7 @@ def shared_dataset(data_xy):
326326
validation_losses = [validate_model(i*batch_size) for i in xrange(n_valid_batches)]
327327
this_validation_loss = numpy.mean(validation_losses)
328328
print('epoch %i, minibatch %i/%i, validation error %f %%' % \
329-
(epoch, minibatch_index+1, n_minibatches, \
329+
(epoch, minibatch_index+1, n_train_batches, \
330330
this_validation_loss*100.))
331331

332332

@@ -347,7 +347,7 @@ def shared_dataset(data_xy):
347347
test_score = numpy.mean(test_losses)
348348
print((' epoch %i, minibatch %i/%i, test error of best '
349349
'model %f %%') %
350-
(epoch, minibatch_index+1, n_minibatches,
350+
(epoch, minibatch_index+1, n_train_batches,
351351
test_score*100.))
352352

353353
if patience <= iter :

0 commit comments

Comments
 (0)