From 2ecf726e6a44fbbf16094f6b8c8a76410d0f29e8 Mon Sep 17 00:00:00 2001
From: Guanghua Yu <742925032@qq.com>
Date: Tue, 10 May 2022 09:44:03 +0800
Subject: [PATCH] fix picofeat assert error (#5911)

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

diff --git a/ppdet/modeling/heads/pico_head.py b/ppdet/modeling/heads/pico_head.py
index ecb4b9764..44f3a214c 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 = []
-- 
GitLab