fix_resnet_vd_nhwc_format

6d42c25c · cuicheng01 · cuicheng01 · 5e814882 · 6d42c25c
隐藏空白更改
内联并排

Showing with 5 addition and 1 deletion

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

未找到文件。
--- a/ppcls/arch/backbone/legendary_models/resnet.py
+++ b/ppcls/arch/backbone/legendary_models/resnet.py
@@ -123,7 +123,11 @@ class ConvBNLayer(TheseusLayer):
        self.is_vd_mode = is_vd_mode
        self.act = act
        self.avg_pool = AvgPool2D(
-            kernel_size=2, stride=stride, padding="SAME", ceil_mode=True)
+            kernel_size=2,
+            stride=stride,
+            padding="SAME",
+            ceil_mode=True,
+            data_format=data_format)
        self.conv = Conv2D(
            in_channels=num_channels,
            out_channels=num_filters,