Skip to content

Custom VOC dataset #173

@sure7018

Description

@sure7018

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions