What is JAX linen (nn) Module?
06/20/2024Pytorch의 nn.Module과 비슷하게 JAX의 neural network library인 FLAX에도 nn.Module이 있는데Pytorch nn.Module과 일대일 대응을 해보려 하다 배운 사실 정리:__init__ -> setupforwad -> __call__ (이건 자기 마음이긴 한데, Pytorch 문법대로 인스턴스를 forward로 쓰고 싶음 이렇게 하는 것임)그런데 setup이 다른 argument를 받지 않는 method라 혼란스러웠는데, nn.Module은 Python 3.7 dataclasses를 가정하고 상속해서 쓰기 때문에 그랬다. 말인즉슨, class variable로 선언해도 인스턴스 variable으로 되기 때문에 __init__ function안 쓰..
Research (연구 관련)
2024. 6. 20. 19:19
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 인터뷰
- Interview
- 피트니스
- 머신러닝
- Docker
- focal length
- VAE
- 2d pose
- Pose2Mesh
- pytorch
- 문경식
- 컴퓨터비젼
- demo
- pyrender
- 헬스
- 에디톨로지
- spin
- densepose
- 비전
- part segmentation
- Transformation
- nerf
- Virtual Camera
- Machine Learning
- deep learning
- camera coordinate
- world coordinate
- 컴퓨터비전
- Generative model
- nohup
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
글 보관함