모델이 학습데이터를 입력받으면 forward 연산을 진행시키는 함수
model 객체를 데이터와 함께 호출하면 자동으로 실행된다.
References
03-05 클래스로 파이토치 모델 구현하기
파이토치의 대부분의 구현체들은 대부분 모델을 생성할 때 클래스(Class)를 사용하고 있습니다. 앞서 배운 선형 회귀를 클래스로 구현해보겠습니다. 앞서 구현한 코드와 다른 점은 …
wikidocs.net
728x90
반응형
'공부' 카테고리의 다른 글
| RAdam (0) | 2023.05.18 |
|---|---|
| assert (0) | 2023.05.18 |
| @staticmethod (0) | 2023.05.18 |
| nn.init.xavier_normal_ (0) | 2023.05.18 |
| GradualWarmupScheduler (0) | 2023.05.18 |
