딥러닝 모델 early stopping 질문

2020.09.17 10:04 3,968 Views

딥러닝할 때 early stopping으로

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

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

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

comment

0 / 1000

isleof

2020.09.17 11:42

tf.keras.callbacks.EarlyStopping의 인자 중에서 restore_best_weights=True 하시면 epoch 중에서
monitor하고 계신 metric 성능이 가장 좋은 모델이 복구됩니다.

팝팝

2020.09.17 12:10

그런 변수가 있었군요... 왜 몰랐지....
감사합니다!!

Current

딥러닝 모델 early stopping 질문

Competition -

Likes 18

Comments 2

5yr ago