run shape dismatch
gentlebreeze1 opened this issue · 0 comments
gentlebreeze1 commented
Size does not match at dimension 0 expected index [128, 1, 4] to be smaller than src [127, 4, 4] apart from dimension 1.
from bbox_pred_select = torch.gather(bbox_pred_view, 1,rois_label.long().view(rois_label.size(0), 1, 1).expand(rois_label.size(0),1, 4)) in fpn.py?
how to deal with this bug?@diamour