change 128 to BATCH_SIZE

8 years ago · 37cd8165b3
parent 1e4566cac0
commit 37cd8165b3
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/v2/framework/tests/mnist.py
+++ b/python/paddle/v2/framework/tests/mnist.py
@ -205,7 +205,8 @@ train_reader = paddle.batch(


 def test(cost_name):
-    test_reader = paddle.batch(paddle.dataset.mnist.test(), batch_size=128)
+    test_reader = paddle.batch(
+        paddle.dataset.mnist.test(), batch_size=BATCH_SIZE)
    cost = []
    error = []
    for data in test_reader():