don't break the no-augmentation case
@@ -102,6 +102,7 @@ def train(sess, loss, x_train, y_train,
xs.append(x)
ys.append(y)
+ if x_batch_preprocessor is not None:
x = x_batch_preprocessor(x)
loss_value = loss.fprop(x, y, **fprop_args)
