ro4577523076/work/aten/src/THC/generic/THCTensorScatterGather.cu:29

File "/home/wanghan/ZWC/FPN.pytorch27/lib/model/fpn/fpn.py", line 252, in forward
    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))
RuntimeError: invalid argument 2: Input tensor must have same size as output tensor apart from the specified dimension at /opt/conda/conda-bld/pytorch_1524577523076/work/aten/src/THC/generic/THCTensorScatterGather.cu:29

posted @ 2020-07-03 21:03  张达嘛  阅读(556)  评论(0)  编辑  收藏  举报