search

SamsungLabs / ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation
MIT License
634 stars 125 forks source link

RuntimeError: Error(s) in loading state_dict for HRNetModel: #16

Open kelisiya opened 3 years ago

kelisiya commented 3 years ago

when I load weights coco_lvis_h18_baseline.pth, the error is size mismatch for maps_transform.0.weight: copying a param with shape torch.Size([16, 2, 1, 1]) from checkpoint, the shape in current model is torch.Size([16, 3, 1, 1]).

HanSeulChung commented 1 year ago

@kelisiya did you solve this problem?? please tell me solution