sshaoshuai / PointRCNN

PointRCNN: 3D Object Proposal Generation and Detection from Point Cloud, CVPR 2019.
MIT License
1.7k stars 425 forks source link

Checkpoint model shape mismatch #209

Open Iven-Wu opened 2 years ago

Iven-Wu commented 2 years ago

I run the eval_rcnn.py with checkpoint downloaded and get this error.

RuntimeError: Error(s) in loading state_dict for PointRCNN: size mismatch for rpn.backbone_net.SA_modules.0.mlps.0.layer0.conv.weight: copying a param with shape torch.Size([16, 3, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 4, 1, 1]). size mismatch for rpn.backbone_net.SA_modules.0.mlps.1.layer0.conv.weight: copying a param with shape torch.Size([32, 3, 1, 1]) from checkpoint, the shape in current model is torch.Size([32, 4, 1, 1]). size mismatch for rpn.backbone_net.FP_modules.0.mlp.layer0.conv.weight: copying a param with shape torch.Size([128, 256, 1, 1]) from checkpoint, the shape in current model is torch.Size([128, 257, 1, 1]).

gehadza commented 1 year ago

Do you solve it ??