파이토치 3.5 클래스로 파이토치 모델 구현하기

앞서 구현한 회귀 모델은 model = nn.Linear(1,1) 이었다 이를 클래스로 구현하면 이렇게 선언할 수 있는데 반드시 숙지할 필요가 있다고 한다 클래스 형태의 모델은 nn.Module을 상속받는다. __init__()에서 모델의 구조와 동작을 정의하는 생성자를 정의한다. 이는 파이썬에서 객체가 갖는 속성값을 초기화하는 역할로, 객체가 생성될 때 자동으로 호출된다. super() 함수를 부르면 여기서 만든 클래스는 nn.Module클래스의 속성을 가지고 초기화 된다. forward() 함수는 모델이 학습데이터를 입력받아서 forward연산을 진행시키는 함수이다. 이 forward()함수는 model 객체를 데이터와 함께 호출하면 자동으로 실행된다. 예를들어 model이란 이름의 객체를 생성한 후, model(입력 데이터)와 같은 형식으로 객체를 호출하면 자동으로 forward 연산이 수행된다. (h(x) 식에 입력 x로부터 예측된 y를 얻는 것을 forward 연산이라고 한...

원문링크 : 파이토치 3.5 클래스로 파이토치 모델 구현하기