RuntimeError about demo!

[Yuejingkun]

❓ Questions and Help

When I run /demo/webcam.py, I met this runtime error

RuntimeError: Error(s) in loading state_dict for GeneralizedRCNN: size mismatch for rpn.head.conv.weight: copying a param with shape torch.Size([256, 256, 3, 3]) from checkpoint, the shape in current model is torch.Size([512, 256, 3, 3]). size mismatch for rpn.head.conv.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([512]). size mismatch for roi_heads.box.feature_extractor.fc6.weight: copying a param with shape torch.Size([1024, 12544]) from checkpoint, the shape in current model is torch.Size([2048, 12544]). size mismatch for roi_heads.box.feature_extractor.fc6.bias: copying a param with shape torch.Size([1024]) from checkpoint, the shape in current model is torch.Size([2048]). size mismatch for roi_heads.box.feature_extractor.fc7.weight: copying a param with shape torch.Size([1024, 1024]) from checkpoint, the shape in current model is torch.Size([2048, 2048]). size mismatch for roi_heads.box.feature_extractor.fc7.bias: copying a param with shape torch.Size([1024]) from checkpoint, the shape in current model is torch.Size([2048]). size mismatch for roi_heads.box.predictor.cls_score.weight: copying a param with shape torch.Size([81, 1024]) from checkpoint, the shape in current model is torch.Size([81, 2048]). size mismatch for roi_heads.box.predictor.bbox_pred.weight: copying a param with shape torch.Size([324, 1024]) from checkpoint, the shape in current model is torch.Size([324, 2048]).

[Yunhao-Liu]

I meet the same question

[MeinhardMark]

Did you solve this?