RuntimeError 之 : CUDA error: device-side assert triggered
生活随笔
收集整理的這篇文章主要介紹了
RuntimeError 之 : CUDA error: device-side assert triggered
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
我們在使用Pytorch時會遇到這種錯誤,這種錯誤往往是因為你的標簽數據里出現了超出標簽范圍的值,比如說你要做100個動作的動作識別,那么你的標簽范圍應該是0,1,2.....99.一般我們采用從0開始的計數方式,但是如果你的標簽數據里出現了諸如-1或100這樣不再此范圍中的數據時,便會報此錯誤。解決辦法就是確認你的標簽的范圍,以及你產生標簽的方式,是不是從1開始計數的。或者一切會導致標簽出錯的原因。
總結
以上是生活随笔為你收集整理的RuntimeError 之 : CUDA error: device-side assert triggered的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: ResNeXt 之 输入数据预处理代码详
- 下一篇: HMM:Hidden Markov Mo