Merge branch 'feature/parallel_nccl_not_support_batchnorm' of https://github.com/reyoung/Paddle into fix-parallel-do-with-non-parameter

7 years ago · c4996fb801
parent f8d0d84f7e 77ff14aa78
commit c4996fb801
1 changed files with 1 additions and 0 deletions
--- a/python/paddle/fluid/tests/unittests/test_parallel_op.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_op.py
@ -170,6 +170,7 @@ class ParallelOpTest(BaseParallelForTest):
        x = fluid.layers.data(shape=[784], dtype='float32', name='img')
        x = yield x
        hidden = fluid.layers.fc(input=x, size=200, param_attr='fc1.w')
+        hidden = fluid.layers.batch_norm(input=hidden)
        loss = fluid.layers.mean(hidden)
        yield loss