未验证 提交 ab3fb4ae 编写于 作者: W wangguanzhong 提交者: GitHub

update BatchNorm2d to BatchNorm2D (#1598)

上级 805dd07a
...@@ -30,7 +30,7 @@ class ConvBNLayer(nn.Layer): ...@@ -30,7 +30,7 @@ class ConvBNLayer(nn.Layer):
weight_attr=ParamAttr(name=name + '.conv.weights'), weight_attr=ParamAttr(name=name + '.conv.weights'),
bias_attr=False) bias_attr=False)
bn_name = name + '.bn' bn_name = name + '.bn'
self.batch_norm = nn.BatchNorm2d( self.batch_norm = nn.BatchNorm2D(
ch_out, ch_out,
weight_attr=ParamAttr( weight_attr=ParamAttr(
name=bn_name + '.scale', regularizer=L2Decay(0.)), name=bn_name + '.scale', regularizer=L2Decay(0.)),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册