fix data format in ppyolo (PaddlePaddle#2403)

kaixin-bai · Mar 22, 2021 · 118e8b7 · 118e8b7
1 parent 6716d44
commit 118e8b7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dygraph/ppdet/modeling/necks/yolo_fpn.py b/dygraph/ppdet/modeling/necks/yolo_fpn.py
@@ -313,6 +313,7 @@ def __init__(self,
 
         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: