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

猜你喜欢

转载自www.cnblogs.com/jiading/p/11979406.html