提交 ec4849c9 编写于 作者: C ceci3

fix init

上级 6b882d42
......@@ -42,14 +42,14 @@ def norm_layer(input,
if norm_type == 'batch_norm':
if affine == True:
param_attr = fluid.ParamAttr(
name=name + '_w', initializer=fluid.initializer.Constant(1.0))
name=name + '_w', initializer=fluid.initializer.Normal(loc=1.0, scale=0.02))
bias_attr = fluid.ParamAttr(
name=name + '_b',
initializer=fluid.initializer.Constant(value=0.0))
else:
param_attr = fluid.ParamAttr(
name=name + '_w',
initializer=fluid.initializer.Constant(1.0),
initializer=fluid.initializer.Normal(loc=1.0, scale=0.02),
trainable=False)
bias_attr = fluid.ParamAttr(
name=name + '_b',
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册