인공지능/Pytorch

딥러닝 모델 Class의 Forward 함수가 궁금했다.

아네스 2021. 1. 9. 00:09
반응형

출처 : Pytorch로 시작하는 딥러닝 입문 책 발췌

자꾸.. Pytorch로 모델링 된 딥러닝 모델을 보면 forward함수를 빼놓을 수 없는데,

위의 model = LinearRegressionModel()로 객체를 생성하고 model.blahblah()로 함수를 부르는게 아니라

model(input_data)이런식으로 부르길레 이게 뭐야 하면서 뇌정지가 왔던 적이 있다.

내부적으로 어떻게 돌아가는진 모르겠지만 생성된 model instance에 train_data를 입력하면 알아서 불리는 메소드라고 하니 이렇게만 일단 이해하고 넘어가자.

반응형