Related to: Machine Learning
개요
PyTorch에서 모든 신경망 레이어와 모델의 기반이 되는 torch.nn.Module 클래스를 설명합니다.
핵심 개념
torch.nn.Module
- ai 모델을 구성하는 layer의 base class
- input, output, forward, backward를 정의해야 함
- 학습 대상이 되는 parameter(tensor)를 정의해야 함
관련 개념
- torch.nn.Parameter은 무엇일까 - Module에 등록되는 학습 파라미터
- PyTorch 딥러닝 학습의 기본 순서 - Module이 사용되는 학습 순서
- 자동 미분(Autograd) - Module의 backward에 사용되는 자동 미분
- optimizer.zero_grad()은 무엇일까 - Module 파라미터 gradient 초기화