/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:115: operator(): block:

GPU上运行神经网络模型,是使用多线程并行加速的,如果出错,真正的错误信息很容易被cuda的错误信息淹没,所以难以看到错误信息,一般引起问题的原因是数组等超限,定位到具体是anchor_ratios=[0.2, 0.5, 1.0, 2.0, 5.0]修改成默认anchor_ratios=[0.5, 1.0, 2.0]就可以啦。

比如,focalloss中gamma数组个数和数据类别数不一致也会报错:

/pytorch/aten/src/ATen/native/cuda/ScatterGatherKernel.cu:115: operator(): block:

将维度修改成一致即可


参考:

pytorch/aten/src/ATen/native/cuda/IndexKernel.cu:53: lambda [](int)->auto::operator()(int)->auto: bl_Wanderer001的博客-CSDN博客

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
上一篇 2022年5月21日 下午3:06
下一篇 2022年5月22日 下午3:00

相关推荐

本站注重文章个人版权,不会主动收集付费或者带有商业版权的文章,如果出现侵权情况只可能是作者后期更改了版权声明,如果出现这种情况请主动联系我们,我们看到会在第一时间删除!本站专注于人工智能高质量优质文章收集,方便各位学者快速找到学习资源,本站收集的文章都会附上文章出处,如果不愿意分享到本平台,我们会第一时间删除!