PointRCNN icon indicating copy to clipboard operation
PointRCNN copied to clipboard

Checkpoint model shape mismatch

Open Iven-Wu opened this issue 2 years ago • 1 comments

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]).

Iven-Wu avatar Mar 24 '22 08:03 Iven-Wu

Do you solve it ??

gehadza avatar Jan 22 '23 18:01 gehadza