In configs/coco/instancesegmentation/maskformer2_R50_bs16_50ep.yaml,
HIDDEN_DIM=256,
My data category is only 1 category
when I load pretrained model
I adjust HIDDEN_DIM,
It always report :
RuntimeError: Error(s) in loading state_dict for MaskFormer:
size mismatch for sem_seg_head.predictor.query_feat.weight: copying a param with shape torch.Size([100, 256]) from checkpoint, the shape in current model is torch.Size([100, 16]).
Is there a way to solve it? Thanks.
(I need to test different hidden dimension)
In configs/coco/instancesegmentation/maskformer2_R50_bs16_50ep.yaml, HIDDEN_DIM=256, My data category is only 1 category when I load pretrained model I adjust HIDDEN_DIM, It always report :
RuntimeError: Error(s) in loading state_dict for MaskFormer: size mismatch for sem_seg_head.predictor.query_feat.weight: copying a param with shape torch.Size([100, 256]) from checkpoint, the shape in current model is torch.Size([100, 16]).
Is there a way to solve it? Thanks. (I need to test different hidden dimension)