Reduce batch size to avoid CUDA_ERROR_OUT_OF_MEMORY

Author: icywang86rui

tests/data/tensorflow_mnist/mnist.py

@@ -174,7 +174,7 @@ def serving_input_fn():
     train_input_fn = tf.estimator.inputs.numpy_input_fn(
         x={"x": train_data},
         y=train_labels,
-        batch_size=100,
+        batch_size=50,
         num_epochs=None,
         shuffle=True)