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


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,所以宽度就超了,修改之后问题就解决了。


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM