未验证 提交 b276610c 编写于 作者: P pk_hk 提交者: GitHub

PPYOLOE fix out_shape, reset return_idx (#6561)

上级 f7890b1b
...@@ -235,7 +235,7 @@ class CSPResNet(nn.Layer): ...@@ -235,7 +235,7 @@ class CSPResNet(nn.Layer):
layers=[3, 6, 6, 3], layers=[3, 6, 6, 3],
channels=[64, 128, 256, 512, 1024], channels=[64, 128, 256, 512, 1024],
act='swish', act='swish',
return_idx=[0, 1, 2, 3, 4], return_idx=[1, 2, 3],
depth_wise=False, depth_wise=False,
use_large_stem=False, use_large_stem=False,
width_mult=1.0, width_mult=1.0,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册