[Error] CUDA error: device-side assert triggered

딥러닝을 하다보면 cuda error를 종종 만나게되는데 정말 화가난다

 

원인 1. Tensor shape 불일치

주로 모델의 출력단계에서 발생.

 

 

원인 2. loss function 입출력 불일치

주로 BCELoss의 경우, input의 범위는 [0, 1]이어야 한다.

그렇지 않으면 에러가 발생한다.

• (solution 2-1) 모델의 출력부분에 시그모이드 등 [0, 1]이 되도록 한다.
• (solution 2-2) 모델의 출력값에 관계없이 사용할 수 있는 BCELogitsLoss()를 사용한다.