From 118e8b75233651205630e9aa3d9fa1cc5944d2a2 Mon Sep 17 00:00:00 2001
From: wangxinxin08 <69842442+wangxinxin08@users.noreply.github.com>
Date: Mon, 22 Mar 2021 21:49:03 +0800
Subject: [PATCH] fix data format in ppyolo (#2403)

---
 dygraph/ppdet/modeling/necks/yolo_fpn.py | 1 +
 1 file changed, 1 insertion(+)

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