PyTorch:훈련

Pywiki
Sam (토론 | 기여)님의 2022년 2월 15일 (화) 11:22 판 (새 문서: model.train()으로 명시하는 이유. 클래스 안에 self.training 이라는 변수가 있어 드롭아웃 옵션이 있을 때 쓰이는데, 실제 작동과 구분하기 위...)
(차이) ← 이전 판 | 최신판 (차이) | 다음 판 → (차이)
둘러보기로 가기 검색하러 가기

model.train()으로 명시하는 이유.

클래스 안에 self.training 이라는 변수가 있어 드롭아웃 옵션이 있을 때 쓰이는데, 실제 작동과 구분하기 위해 모델의 사용 때마다 구분해주어야 한다.(이런 부분은 충분히 자동화가 가능할텐데 불편하지.)