[오류]RuntimeError: CUDA error: device-side assert triggered

👩‍💻/NLP

글로랴 2024. 1. 24. 16:35

[pytorch] device-side assert triggered error 해결방법

1. 에러 발생 상황 torchvision.model 에서 resnet 50을 불러와서 fully connected 부분을 변경시킨 모델을 응용하여 새로운 모델을 작성중에 에러가 발생되었습니다. 에러 내용은 다음과 같습니다. RuntimeError

mopipe.tistory.com

클래스 인덱스는 " 0 " 부터 시작해야 한다는 거....