- 분류 전체보기 (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
- matplotlib
- 텝스공부
- 논문작성법
- 인공지능
- 텝스
- 수치해석
- 우분투
- 논문작성
- obsidian
- MATLAB
- IEEE
- pytorch
- Zotero
- Statics
- Linear algebra
- Dear abby
- 수식삽입
- 딥러닝
- Numerical Analysis
- WOX
- teps
- 에러기록
- Julia
- JAX
- 고체역학
- LaTeX
- Python
- 옵시디언
- ChatGPT
- 생산성
Archives
- Today
- Total
뛰는 놈 위에 나는 공대생
[JAX] JAX에서 gradient 추척을 멈추는 방법 본문
다음과 같이 jax.lax.stop_gradient로 묶어준 결과를 사용하면 그 이전까지는 gradient가 기록되지 않는다.
jax.lax.stop_gradient(sol.ts)'연구 Research > 인공지능 Artificial Intelligent' 카테고리의 다른 글
| [JAX] vmap과 jit의 속도 (0) | 2023.09.20 |
|---|---|
| [인공지능] 인공지능 라이브러리 정리 (0) | 2023.08.24 |
| [PyTorch] 인공지능 재현성을 위한 설정과 주의할 점 (0) | 2023.08.11 |
| Neural networks의 convergence, convexity에 대한 논문 (0) | 2023.07.31 |
| [JAX] JAX 기반 Neural ODE 라이브러리 : diffrax (0) | 2023.07.28 |
'연구 Research/인공지능 Artificial Intelligent' Related Articles
more
Comments