RuntimeError: CUDA error: device-side assert triggered

segmentation train model을 돌리는데 계속 위와 같은 RuntimeError가 뜬다.. dataset 관련 문제 같은데 정확하게 뭔지 모르겠다..

 

구글링해보니 cuda상에서 발생하는 문제라 "cuda:0"을 "cpu"로 바꿔서 돌려보니까 아래 사진과 같은 에러가 떴다

RuntimeError: index 96 is out of bounds for dimension 1 with size 2 라는데.. 

음 어캐 고쳐야하지