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,所以寬度就超了,修改之后問題就解決了。