CV 기초대회 5강 - Model 1 - Model with Pytorch

Design Model with Pytorch

Pytorch

1) Low-level

2) Pythonic

3) Flexibility

• low level이기 때문에 기능 변경 등에 용이함

• 위의 두 사진 중 우측과 유사

• Keras 등 다른 라이브러리에 비해 코드가 길고 직관적이지 않지만, pythonic하고 low-level이기 때문에 내용을 이해한다면 활용에 유리함

nn.Module

• pytorch 모델의 모든 레이어는 nn.Module 클래스를 상속함

• 내 모델을 정의하는 순간 그 모델에 연결된 모든 module을 확인 가능

• 모든 nn.Module은 forward() 함수를 가지므로 custom model의 forward() 실행 시 그 모델 내부의 모듈 각각의 forward가 실행됨

Parameters

.state_dict()

• key + parameter(dictionary)
  

.parameters()

• parameter
  

data, grad, requires_grad

• 각각 파라미터 값/gradient/gradient 필요 여부