segmentation train model을 돌리는데 계속 위와 같은 RuntimeError가 뜬다.. dataset 관련 문제 같은데 정확하게 뭔지 모르겠다..
구글링해보니 cuda상에서 발생하는 문제라 "cuda:0"을 "cpu"로 바꿔서 돌려보니까 아래 사진과 같은 에러가 떴다
RuntimeError: index 96 is out of bounds for dimension 1 with size 2 라는데..
음 어캐 고쳐야하지
'파이썬' 카테고리의 다른 글
dicom to nifti에서 발생하는 오류 - TypeError: float() argument must be a string or a number, not 'NoneType' (0) | 2022.05.01 |
---|