提交 c4996fb8 编写于 作者: Q qiaolongfei

Merge branch 'feature/parallel_nccl_not_support_batchnorm' of...

Merge branch 'feature/parallel_nccl_not_support_batchnorm' of https://github.com/reyoung/Paddle into fix-parallel-do-with-non-parameter
...@@ -170,6 +170,7 @@ class ParallelOpTest(BaseParallelForTest): ...@@ -170,6 +170,7 @@ class ParallelOpTest(BaseParallelForTest):
x = fluid.layers.data(shape=[784], dtype='float32', name='img') x = fluid.layers.data(shape=[784], dtype='float32', name='img')
x = yield x x = yield x
hidden = fluid.layers.fc(input=x, size=200, param_attr='fc1.w') hidden = fluid.layers.fc(input=x, size=200, param_attr='fc1.w')
hidden = fluid.layers.batch_norm(input=hidden)
loss = fluid.layers.mean(hidden) loss = fluid.layers.mean(hidden)
yield loss yield loss
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册