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