[colab] cuda error device-side assert triggered

모델 학습 후, 예측을 진행하려 하니 cuda error device-side assert triggered 오류가 나왔다.

 

구글링 결과 batch 사이즈를 줄이라고 했는데, 내 경우에는 줄여도 되지 않았다.

다른 글을 찾아보니, class의 불균형 문제 때문에 그럴 수도 있다고 한다.

 

python의 imblearn 라이브러리를 활용해서

oversampling이나 undersampling을 통해 class의 개수를 비슷하게 맞춰주니 잘 작동했다.