diff --git a/ppdet/modeling/heads/pico_head.py b/ppdet/modeling/heads/pico_head.py index ecb4b976403ef18ee5174a80885bcc8015707967..44f3a214cc8d467308fc1861f1e30e61377d9c79 100644 --- a/ppdet/modeling/heads/pico_head.py +++ b/ppdet/modeling/heads/pico_head.py @@ -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 = []