smd
smd copied to clipboard
Unable load custom trained faster rcnn 50 fpn model
RuntimeError: Error(s) in loading state_dict for FasterRCNN: size mismatch for roi_heads.box_predictor.cls_score.weight: copying a param with shape torch.Size([11, 1024]) from checkpoint, the shape in current model is torch.Size([10, 1024]). size mismatch for roi_heads.box_predictor.cls_score.bias: copying a param with shape torch.Size([11]) from checkpoint, the shape in current model is torch.Size([10]). size mismatch for roi_heads.box_predictor.bbox_pred.weight: copying a param with shape torch.Size([44, 1024]) from checkpoint, the shape in current model is torch.Size([40, 1024]). size mismatch for roi_heads.box_predictor.bbox_pred.bias: copying a param with shape torch.Size([44]) from checkpoint, the shape in current model is torch.Size([40]).