PointRCNN
PointRCNN copied to clipboard
Checkpoint model shape mismatch
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]).
Do you solve it ??