首页 > 其他 > 详细

Pytorch报错:cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

时间:2019-12-03 21:34:51      阅读:130      评论:0      收藏:0      [点我收藏+]

Pytorch报错:cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

这种问题是网上比较常见的,一般的原因就是Label没有从0开始导致数组或者tensor超出范围。我这次也是这个原因,具体来说,是由于使用了nll_loss造成的。关于NLLLoss,可以看我的这篇文章

在计算NLLLoss时,要算X_label,但是输入的label是从1到10,而X的宽度就是10,所以宽度就超了,修改之后问题就解决了。

Pytorch报错:cuda runtime error (59) : device-side assert triggered at /pytorch/aten/src/THC/generic/THCTensorMath.cu:26

原文:https://www.cnblogs.com/jiading/p/11979406.html

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