- 분류 전체보기 (434)
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 |
Tags
- pytorch
- 생산성
- 텝스
- 인공지능
- 텝스공부
- matplotlib
- 에러기록
- 우분투
- 수식삽입
- teps
- 옵시디언
- Statics
- Julia
- LaTeX
- 딥러닝
- IEEE
- 논문작성
- MATLAB
- 수치해석
- ChatGPT
- 고체역학
- JAX
- Dear abby
- WOX
- obsidian
- Zotero
- Python
- Linear algebra
- Numerical Analysis
- 논문작성법
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