Not support batchnorm

tonyyang-svail-patch-1
Yu Yang 7 years ago
parent f0548679b3
commit 77ff14aa78

@ -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

Loading…
Cancel
Save