fix batchnorm2d data_format

9f049f11 · zhiboniu · 939a35d6 · 9f049f11
隐藏空白更改
内联并排

Showing with 4 addition and 1 deletion

ppcls/arch/backbone/legendary_models/resnet.py ppcls/arch/backbone/legendary_models/resnet.py +4 -1

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/resnet.py
+++ b/ppcls/arch/backbone/legendary_models/resnet.py
@@ -138,7 +138,10 @@ class ConvBNLayer(TheseusLayer):
        bias_attr = ParamAttr(learning_rate=lr_mult, trainable=True)

        self.bn = BatchNorm2D(
-            num_filters, weight_attr=weight_attr, bias_attr=bias_attr)
+            num_filters,
+            weight_attr=ParamAttr(learning_rate=lr_mult),
+            bias_attr=ParamAttr(learning_rate=lr_mult),
+            data_format=data_format)
        self.relu = nn.ReLU()

    def forward(self, x):