Error handling

2021. 5. 31. 01:32공부한 것 정리/개발환경설정

반응형

RuntimeError: CUDA error: device-side assert triggered

1. /pytorch/aten/src/THCUNN/ClassNLLCriterion.cu:108: cunn_ClassNLLCriterion_updateOutput_kernel: block: [0,0,0], thread: [0,0,0] Assertion `t >= 0 && t < n_classes` failed
     -> 데이터셋 정의에서 -1값이 들어가면 오류 발생

          => t >= 0 && t < n_classes 으로 바꿔줘야함

     -> 혹은 CE 사용할 때 pred 인풋값이 인지 확인 // index_ignore = 해당인덱스 추가하기

 

반응형