reverse to BatchNorm

7f0075e8 · zhiboniu · 9f049f11 · 7f0075e8
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

ppcls/arch/backbone/legendary_models/resnet.py ppcls/arch/backbone/legendary_models/resnet.py +3 -3

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

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

    def forward(self, x):