딥러닝 모델 early stopping 질문

댓글 2

딥러닝할 때 early stopping으로


일정 성능 이상 나오지 않으면 epoch가 중단 되잖아요


그럼 model은 자동적으로 마지막 모델이 저장되는 건가요?


아니면 성능이 가장 좋은 것이 저장되는 건가요??

댓글 2개
  • isleof isleof 2020.09.17 11:42
    tf.keras.callbacks.EarlyStopping의 인자 중에서 restore_best_weights=True 하시면 epoch 중에서
    monitor하고 계신 metric 성능이 가장 좋은 모델이 복구됩니다.
  • 팝팝 팝팝 2020.09.17 12:10
    그런 변수가 있었군요... 왜 몰랐지....
    감사합니다!!
로그인이 필요합니다로그인이 필요합니다
목록으로