[PyTorch] model.train()

clean·2023년 11월 3일
0

model.train() 함수

모델을 '학습 모드'로 만드는 메소드이다.
모델을 학습 모드로 설정을 함으로써, 파라미터 업데이트 및 그래이디언트 계산을 가능하게 한다.
따라서 모델을 학습하기 전에 호출을 한다.

반대로 model.eval() 함수를 호출하면, 모델이 '평가 모드'로 바뀌어서 파라미터가 업데이트 되지 않는 것이다.

profile
' v ' 잔디 심는 사람

0개의 댓글