-
Notifications
You must be signed in to change notification settings - Fork 469
Description
When I use a custom VOC format dataset, because the category is only 14 and does not match the weight of the pre training, the following error will be reported. How can I modify it??
size mismatch for cls_score_net.weight: copying a param with shape torch.Size([1001, 2048]) from checkpoint, the shape in current model is torch.Size([15, 2048]).
size mismatch for cls_score_net.bias: copying a param with shape torch.Size([1001]) from checkpoint, the shape in current model is torch.Size([15]).
size mismatch for bbox_pred_net.weight: copying a param with shape torch.Size([4004, 2048]) from checkpoint, the shape in current model is torch.Size([60, 2048]).
size mismatch for bbox_pred_net.bias: copying a param with shape torch.Size([4004]) from checkpoint, the shape in current model is torch.Size([60]).