실제 torch weights 살펴보기
기술/PyTorch

실제 torch weights 살펴보기

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