Related to: Machine Learning
개요
PyTorch에서 모델의 학습 대상 파라미터를 표현하는 torch.nn.Parameter 클래스를 설명합니다.
핵심 개념
torch.nn.Parameter
- Tensor를 상속받는 Class
- nn.Module 내에 Attribute가 될 때,
required_grad=True로 지정되어 자동으로 학습대상이 됨 - 일반 Tensor와 달리, Module에 등록되어
model.parameters()로 조회 가능
관련 개념
- torch.nn.Module은 무엇일까 - Parameter가 속하는 nn.Module 클래스
- torch.tensor의 requires_grad param의 기능 - requires_grad 파라미터의 역할
- 자동 미분(Autograd) - Parameter에 자동 적용되는 미분 메커니즘
- optimizer.zero_grad()은 무엇일까 - Parameter의 grad를 초기화하는 함수