``` n_train_batches = train_set_x.get_value(borrow=True).shape[0] / batch_size n_valid_batches = valid_set_x.get_value(borrow=True).shape[0] / batch_size n_test_batches = test_set_x.get_value(borrow=True).shape[0] / batch_size ``` should be: ``` n_train_batches = train_set_x.get_value(borrow=True).shape[0] / batch_size + 1 n_valid_batches = valid_set_x.get_value(borrow=True).shape[0] / batch_size + 1 n_test_batches = test_set_x.get_value(borrow=True).shape[0] / batch_size + 1 ```