PPYOLOE fix out_shape, reset return_idx (#6561)

b276610c · pk_hk · GitHub · f7890b1b · b276610c
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppdet/modeling/backbones/cspresnet.py ppdet/modeling/backbones/cspresnet.py +1 -1

未找到文件。
--- 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,