728x90
반응형
반응형
- 전체 에러 문구
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) 차원의 크기를 설정했을 때
에도 발생할 수 있는 에러입니다.
728x90
반응형