From 37cd8165b3089c8e4a6ce743f5e0ee8c029ba46b Mon Sep 17 00:00:00 2001 From: qiaolongfei 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 eefd5709a3b..e878bfa4e9b 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(): -- GitLab