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 = 해당인덱스 추가하기
반응형