diff --git a/dygraph/ppdet/modeling/necks/yolo_fpn.py b/dygraph/ppdet/modeling/necks/yolo_fpn.py
index 77b6d885df71c105dc391cee6782ff7b863c2673..456bfae2097f3187ced647367986d20fee2730a5 100644
--- 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: