From b276610c3da6d2c46532549b7a08600733db9909 Mon Sep 17 00:00:00 2001
From: pk_hk <82303451+pkhk-1@users.noreply.github.com>
Date: Wed, 3 Aug 2022 14:34:53 +0800
Subject: [PATCH] PPYOLOE fix out_shape, reset return_idx (#6561)

---
 ppdet/modeling/backbones/cspresnet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/ppdet/modeling/backbones/cspresnet.py b/ppdet/modeling/backbones/cspresnet.py
index 833dd43c3..5268ec835 100644
--- a/ppdet/modeling/backbones/cspresnet.py
+++ b/ppdet/modeling/backbones/cspresnet.py
@@ -235,7 +235,7 @@ class CSPResNet(nn.Layer):
                  layers=[3, 6, 6, 3],
                  channels=[64, 128, 256, 512, 1024],
                  act='swish',
-                 return_idx=[0, 1, 2, 3, 4],
+                 return_idx=[1, 2, 3],
                  depth_wise=False,
                  use_large_stem=False,
                  width_mult=1.0,
-- 
GitLab