From ab3fb4ae7d53280c07c6c230c69feff7224517de Mon Sep 17 00:00:00 2001 From: wangguanzhong Date: Mon, 26 Oct 2020 10:55:57 +0800 Subject: [PATCH] update BatchNorm2d to BatchNorm2D (#1598) --- ppdet/modeling/backbone/darknet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppdet/modeling/backbone/darknet.py b/ppdet/modeling/backbone/darknet.py index cb5b41002..b1e386cae 100755 --- a/ppdet/modeling/backbone/darknet.py +++ b/ppdet/modeling/backbone/darknet.py @@ -30,7 +30,7 @@ class ConvBNLayer(nn.Layer): weight_attr=ParamAttr(name=name + '.conv.weights'), bias_attr=False) bn_name = name + '.bn' - self.batch_norm = nn.BatchNorm2d( + self.batch_norm = nn.BatchNorm2D( ch_out, weight_attr=ParamAttr( name=bn_name + '.scale', regularizer=L2Decay(0.)), -- GitLab