Use mmcv.runner.ModuleList instead of nn.ModuleList in mmdet3d.models.dense_heads.centerpoint_head

[anti-destiny]

As you can see from here:

mmdetection3d/mmdet3d/models/dense_heads/centerpoint_head.py

Line 321 in 14c5ded

self.task_heads = nn.ModuleList()

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.

[ZCMax]

It's indeed a small bug in CenterPoint head, we'll fix it and re-train the model to see the performance improvement.

[42nick]

Hi,
how is the retraining going?

The problem is still there.