PyTorch:훈련

Pywiki
둘러보기로 가기 검색하러 가기

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

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