Skip to content

Commit c4996fb

Browse files
committed
Merge branch 'feature/parallel_nccl_not_support_batchnorm' of https://github.com/reyoung/Paddle into fix-parallel-do-with-non-parameter
2 parents f8d0d84 + 77ff14a commit c4996fb

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

python/paddle/fluid/tests/unittests/test_parallel_op.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,7 @@ def __network__():
170170
x = fluid.layers.data(shape=[784], dtype='float32', name='img')
171171
x = yield x
172172
hidden = fluid.layers.fc(input=x, size=200, param_attr='fc1.w')
173+
hidden = fluid.layers.batch_norm(input=hidden)
173174
loss = fluid.layers.mean(hidden)
174175
yield loss
175176

0 commit comments

Comments
 (0)