컴퓨터비전관련

Loss nan 값 나올 때

감._.자 2022. 9. 27. 15:41
728x90
반응형
torch.exp(x)

이때 x 값이 너무 큰 값이 들어가면 숫자가 아니라 inf 로 결과가 나와버림.

 

inf 값에 다른 수치를 곱하거나 나눠버리면 nan값으로 변하게 되어서 loss가 nan으로 나오게 되는거임

 

x를 특정 값으로 나눠서 수치를 작게 만들거나 해주면 잘 해결됨

 

728x90
반응형