딥러닝 모델 early stopping 질문

2020.09.17 10:04 2,944 조회

딥러닝할 때 early stopping으로


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


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


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

로그인이 필요합니다
0 / 1000
isleof
2020.09.17 11:42

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

팝팝
2020.09.17 12:10

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