"PyTorch:훈련"의 두 판 사이의 차이

Pywiki
둘러보기로 가기 검색하러 가기
(새 문서: model.train()으로 명시하는 이유. 클래스 안에 self.training 이라는 변수가 있어 드롭아웃 옵션이 있을 때 쓰이는데, 실제 작동과 구분하기 위...)
 
(차이 없음)

2022년 2월 15일 (화) 11:22 기준 최신판

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

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