반응형
에러 메시지:
/opt/conda/conda-bld/pytorch_1682343995026/work/aten/src/aten/native/cuda/loss.cu:240: nll_loss_forward_reduce_cuda_kernel_2d: block: [0,0,0], thread: [0,0,0] assertion `t >= 0 && t < n_classes` failed.
에러 메시지 해석:
loss를 계산하는 과정에서 label(ground_truth)과 output(predict_truth)의 클래스 숫자가 예상 밖인 0~n_classes-1 이외의 숫자로 나올 경우 해당 오류가 발생한다.
해결:
필자의 경우 사용하는 데이터셋 중 라벨이 없는 데이터셋이 있는데, 해당 데이터셋을 사용하여 label 부분을 print 한 결과 잘못된 label이 -1로 전부 나와 해당 오류를 확인했다. 따라서, 적절한 데이터셋으로 바꿔 다시 코드를 실행했고, 적절한 label을 얻어 해당 오류가 해결되었다.
반응형