torch.nn.Linear
https://pytorch.org/docs/stable/_modules/torch/nn/modules/linear.html#Linear
https://pytorch.org/docs/stable/generated/torch.nn.functional.linear.html
https://pytorch.org/docs/stable/_modules/torch/nn/modules/module.html#Module
Module에 내가 건드려봤던 register hook들도 있음. 그 외에 zero_grad, requires_grad_, children 등등..
https://doodlrudco.tistory.com/entry/torchnnParameter-%EC%97%90-%EA%B4%80%ED%95%B4%EC%84%9C
nn.Parameter는 일종의 저장소 역할이라고 보면 될것 같다. 변수 저장소.
'기술 > PyTorch' 카테고리의 다른 글
nn.AdaptiveAvgPool1d (0) | 2022.02.10 |
---|---|
DistributedDataParallel 관련 링크 정리 (0) | 2021.12.27 |
torchvision.models 의 IntermediateLayerGetter (1) | 2021.10.06 |
pbar 고급진 사용법 (0) | 2021.09.13 |
nn.Sequential 에 OrderedDict 넣어줘도 됨 (0) | 2021.08.18 |