Deep Learning
[Deep Learning] Pytorch의 nn.Module 이해하기
nn.Module을 상속받는 클래스에서의 forward() 함수 토치에서 모델을 만들 때, 다음과 같은 형태로 모델을 만든다. PyTorch 내장 모델과 사용자 정의 모델 모두 이 형태로 만들어야 한다. 1. torch.nn.Module을 상속받아야 한다. 2. __init__()과 forward()를 override 해줘야 한다. __init__()에서는 사용될 모듈, 활성화 함수 등을 정의한다. forward()에서는 모델에서 실행되어야 하는 계산을 정의한다. backward 계산은 backward()를 이용하면 알아서 해주기 때문에 forward()만 정의하면 된다. input을 넣고 어떤 과정을 거쳐 output이 나올지를 정의해 준다는 느낌이다. class GRU(nn.Module): def _..
