Notice
Recent Posts
Recent Comments
Link
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | 3 | 4 | 5 | 6 | 7 |
| 8 | 9 | 10 | 11 | 12 | 13 | 14 |
| 15 | 16 | 17 | 18 | 19 | 20 | 21 |
| 22 | 23 | 24 | 25 | 26 | 27 | 28 |
| 29 | 30 | 31 |
Tags
- teps
- Python
- IEEE
- JAX
- 수식삽입
- 논문작성
- Numerical Analysis
- pytorch
- 옵시디언
- 고체역학
- 딥러닝
- obsidian
- ChatGPT
- Statics
- Julia
- WOX
- 텝스공부
- 에러기록
- 텝스
- 논문작성법
- matplotlib
- 수치해석
- LaTeX
- Linear algebra
- 인공지능
- Zotero
- 생산성
- 우분투
- Dear abby
- MATLAB
Archives
- Today
- Total
뛰는 놈 위에 나는 공대생
[JAX] 학습 중 NaN 값이 나올 때 찾는 방법 본문
JAX로 학습하는 도중에 NAN값이 나와서 어디부터 원인인지 찾기가 어려웠다. 이럴 때는 아래 코드를 추가하면 된다.
from jax.config import config
config.update("jax_debug_nans", True)
이렇게 할 경우에 NAN이 발생하는 즉시 어떤 코드에서 문제가 발생하는지를 알려주고 코드가 종료된다.
'연구 Research > 인공지능 Artificial Intelligent' 카테고리의 다른 글
| [JAX] device 확인, default device 설정 (0) | 2023.04.13 |
|---|---|
| [JAX] JAX와 Torch, CUDA, cudnn 버전 맞추기 (0) | 2023.04.12 |
| [JAX] Gaussian process 파라미터에 따른 결과 visualization (0) | 2023.03.24 |
| [JAX] 기본 Neural Networks 모델 (0) | 2023.03.23 |
| [JAX] Gradient, Jacobian, Hessian 등 미분값 구하기 (2) | 2023.03.22 |
'연구 Research/인공지능 Artificial Intelligent' Related Articles
more
Comments