@@ -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