change 128 to BATCH_SIZE

37cd8165 · qiaolongfei · 1e4566ca · 37cd8165
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

python/paddle/v2/framework/tests/mnist.py python/paddle/v2/framework/tests/mnist.py +2 -1

未找到文件。
--- 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():