파이썬
RuntimeError: CUDA error: device-side assert triggered
밀루루
2022. 5. 24. 11:17
segmentation train model을 돌리는데 계속 위와 같은 RuntimeError가 뜬다.. dataset 관련 문제 같은데 정확하게 뭔지 모르겠다..
구글링해보니 cuda상에서 발생하는 문제라 "cuda:0"을 "cpu"로 바꿔서 돌려보니까 아래 사진과 같은 에러가 떴다
RuntimeError: index 96 is out of bounds for dimension 1 with size 2 라는데..
음 어캐 고쳐야하지