search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 544 forks source link

ResNext models doesnt works #66

Closed shridharkini6 closed 6 years ago

shridharkini6 commented 6 years ago

getting this error when run with e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x model. load_detectron_weight(maskRCNN, load_detectron) File "/datadrive/Shridhar/detectron/mask-rcnn.pytorch/lib/utils/detectron_weight_helper.py", line 22, in load_detectron_weight p_tensor.copy_(torch.Tensor(src_blobs[d_name])) RuntimeError: The expanded size of the tensor (1024) must match the existing size (256) at non-singleton dimension 0

Where as ResNet model works flawlesly. Does it supports ResNext models so far? Thanks

roytseng-tw commented 6 years ago

Hi, I have made ResNeXt backbones work on the latest master. Remember to download the pretrained weights for ResNeXt.