未验证 提交 2ecf726e 编写于 作者: G Guanghua Yu 提交者: GitHub

fix picofeat assert error (#5911)

上级 2c3225f8
......@@ -91,7 +91,7 @@ class PicoFeat(nn.Layer):
self.reg_convs = []
if use_se:
assert share_cls_reg == True, \
'In the case of using se, share_cls_reg is not supported'
'In the case of using se, share_cls_reg must be set to True'
self.se = nn.LayerList()
for stage_idx in range(num_fpn_stride):
cls_subnet_convs = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册