From c064334a041fe528cbca24ae0a892fc0e0905643 Mon Sep 17 00:00:00 2001
From: wjm <897383984@qq.com>
Date: Tue, 30 Aug 2022 14:13:07 +0800
Subject: [PATCH] fix_fpn_senet (#6791) (#6794)

---
 ppdet/modeling/backbones/senet.py | 2 ++
 ppdet/modeling/necks/fpn.py       | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/ppdet/modeling/backbones/senet.py b/ppdet/modeling/backbones/senet.py
index eb0bad33f..db1e29b97 100644
--- a/ppdet/modeling/backbones/senet.py
+++ b/ppdet/modeling/backbones/senet.py
@@ -16,6 +16,8 @@ import paddle.nn as nn
 
 from ppdet.core.workspace import register, serializable
 from .resnet import ResNet, Blocks, BasicBlock, BottleNeck
+from ..shape_spec import ShapeSpec
+from .name_adapter import NameAdapter
 
 __all__ = ['SENet', 'SERes5Head']
 
diff --git a/ppdet/modeling/necks/fpn.py b/ppdet/modeling/necks/fpn.py
index 0633fb5b2..d08ca415c 100644
--- a/ppdet/modeling/necks/fpn.py
+++ b/ppdet/modeling/necks/fpn.py
@@ -33,7 +33,7 @@ class FPN(nn.Layer):
     Args:
         in_channels (list[int]): input channels of each level which can be 
             derived from the output shape of backbone by from_config
-        out_channel (list[int]): output channel of each level
+        out_channel (int): output channel of each level
         spatial_scales (list[float]): the spatial scales between input feature
             maps and original input image which can be derived from the output 
             shape of backbone by from_config
-- 
GitLab