좌충우돌

pytorch NaN 발생 위치 찾기

학습 스크립트 중 가급적 처음 시기에 아래 기능을 활성화 해 준다.

NaN이 발생한 경우, backward 시점에서 스크립트의 어느 줄에서 발생했는지 출력해주며, 그 이유도 간접적으로 알 수 있다.

정상적으로 NaN에 대하여 예외 처리를 하였더라도 NaN이 발생한 것으로 간주하여 스크립트 실행이 중단되게 되므로 유의하여야 한다.


Tags:
Add a Comment Trackback