首页 > 其他 > 详细

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

时间:2020-07-03 22:54:17      阅读:186      评论:0      收藏:0      [点我收藏+]

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!