From 37cd8165b3089c8e4a6ce743f5e0ee8c029ba46b Mon Sep 17 00:00:00 2001
From: qiaolongfei <qiaolongfei@baidu.com>
Date: Wed, 23 Aug 2017 10:56:56 -0700
Subject: [PATCH] change 128 to BATCH_SIZE
---
python/paddle/v2/framework/tests/mnist.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/python/paddle/v2/framework/tests/mnist.py b/python/paddle/v2/framework/tests/mnist.py
index eefd5709a3..e878bfa4e9 100644
--- 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():