Update batch_size

7 years ago · 83c72536e6
parent dc488c17d1
commit 83c72536e6
1 changed files with 2 additions and 1 deletions
--- 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')