[Pytorch] RuntimeError: CUDA error: device-side assert triggered

반응형

- 전체 에러 문구

RuntimeError: CUDA error: device-side assert triggered

​/pytorch/aten/src/ATen/native/cuda/Indexing.cu:699: indexSelectLargeIndex: block: [129,0,0], thread: [127,0,0] Assertion `srcIndex < srcSelectDimSize` failed.

​

​

Pytorch의 nn.embedding() 레이어에 유효하지 않은 값(-1 이나 nan)이 입력으로 들어갔을 때 발생하는 에러입니다. 저의 경우에는 -1이 입력으로 들어가 발생하였습니다.

​

​

- 해결 방법

유효한 입력값이 들어갈 수 있도록 수정해주었습니다.

​

이외에도 아래의 경우에 에러가 발생할 수 있습니다!

- 실제 데이터셋과 다르게 출력(output) 차원의 크기를 설정했을 때와

- 실제 데이터셋과 다르게 입력(input) 차원의 크기를 설정했을 때

에도 발생할 수 있는 에러입니다.

​