mmdetection3d
mmdetection3d copied to clipboard
Use mmcv.runner.ModuleList instead of nn.ModuleList in mmdet3d.models.dense_heads.centerpoint_head
As you can see from here: https://github.com/open-mmlab/mmdetection3d/blob/14c5ded49dd5df7a52368bd54c32a003c2521e1d/mmdet3d/models/dense_heads/centerpoint_head.py#L321
task_heads is initialized as a nn.ModuleList
However, nn.ModuleList does not implement the init_weights function which is declared in the BaseModule class.
So, you should replaced nn.ModuleList with mmcv.runner.ModuleList. Otherwise, the CenterPointHead won't be correctly initialized.
It's indeed a small bug in CenterPoint head, we'll fix it and re-train the model to see the performance improvement.
Hi, how is the retraining going?
The problem is still there.