提交 f1fe166c 编写于 作者: W wanghaoshuang

Fix batch normal error

上级 3deecd96
......@@ -18,8 +18,8 @@ def conv_bn_pool(input,
param_attr=param if param_0 is None else param_0,
act=None, # LinearActivation
use_cudnn=True)
tmp = fluid.layers.batch_norm(
input=tmp, act=act, param_attr=param, bias_attr=bias)
tmp = fluid.layers.batch_norm(
input=tmp, act=act, param_attr=param, bias_attr=bias)
tmp = fluid.layers.pool2d(
input=tmp, pool_size=2, pool_type='max', pool_stride=2, use_cudnn=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册