提交 77ff14aa 编写于 作者: Y Yu Yang

Not support batchnorm

上级 f0548679
...@@ -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.
先完成此消息的编辑!
想要评论请 注册