diff --git a/ppdet/modeling/heads/pico_head.py b/ppdet/modeling/heads/pico_head.py index 07cc3c7ca7c93710bd41d49d810cbd6816bf3739..814d7d4e6159d330e1f2a99dc81c35aaf2ace97a 100644 --- a/ppdet/modeling/heads/pico_head.py +++ b/ppdet/modeling/heads/pico_head.py @@ -499,8 +499,8 @@ class PicoHeadV2(GFLHead): self.gfl_head_reg = None self.scales_regs = None - self.head_cls_list = [] - self.head_reg_list = [] + self.head_cls_list = nn.LayerList() + self.head_reg_list = nn.LayerList() self.cls_align = nn.LayerList() for i in range(len(fpn_stride)):