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
ro4577523076/work/aten/src/THC/generic/THCTensorScatterGather.cu:29
原文:https://www.cnblogs.com/duzetao/p/13232661.html