From 83c72536e662b5e8be7eb1a3e69a4f02ac584b20 Mon Sep 17 00:00:00 2001
From: Yang Yu <yuyang18@baidu.com>
Date: Thu, 11 Jan 2018 12:35:18 +0800
Subject: [PATCH] Update batch_size

---
 python/paddle/v2/fluid/tests/test_parallel_op.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/paddle/v2/fluid/tests/test_parallel_op.py b/python/paddle/v2/fluid/tests/test_parallel_op.py
index dde7206e4f..2b51a1f504 100644
--- a/python/paddle/v2/fluid/tests/test_parallel_op.py
+++ b/python/paddle/v2/fluid/tests/test_parallel_op.py
@@ -147,7 +147,8 @@ class ParallelOpTest(BaseParallelForTest):
         self.run_test(
             callback=__network__,
             feed={
-                'img': numpy.random.random(size=(128, 784)).astype('float32')
+                'img':
+                numpy.random.random(size=(128 * 3, 784)).astype('float32')
             },
             fetch='fc1.w@GRAD')
 
-- 
GitLab