fix data format in ppyolo (#2403)

118e8b75 · wangxinxin08 · GitHub · 6716d44c · 118e8b75
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

dygraph/ppdet/modeling/necks/yolo_fpn.py dygraph/ppdet/modeling/necks/yolo_fpn.py +1 -0

未找到文件。
--- a/dygraph/ppdet/modeling/necks/yolo_fpn.py
+++ b/dygraph/ppdet/modeling/necks/yolo_fpn.py
@@ -313,6 +313,7 @@ class PPYOLOFPN(nn.Layer):

        self.spp = kwargs.get('spp', False)
        self.conv_block_num = kwargs.get('conv_block_num', 2)
+        self.data_format = data_format
        if self.coord_conv:
            ConvLayer = CoordConv
        else: