모든 시대 이후에 모델을 저장하는 Keras 콜백 예제?


17

누군가가 각 시대마다 모델을 저장하기 위해 콜백을 사용하여 Keras의 간단한 예를 게시 할 수 있습니까? 웨이트 절약의 예를 찾을 수 있지만 모든 훈련 시대마다 완벽하게 작동하는 모델을 저장할 수 있기를 원합니다.

답변:


22

Keras 콜백 'ModelCheckpoint' 에서 'save_weights_only'를 False로 설정 하면 전체 모델이 저장됩니다. 위의 링크에서 가져온이 예제는 성능에 관계없이 모든 시대를 전체 모델로 저장합니다.

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

개선 된 모델 만 저장하고 저장된 모델을로드하는 등 몇 가지 예가 여기에 있습니다 .


10

파일 경로에 epoch 변수를 포함시켜야합니다. 그렇지 않으면 저장된 모델은 모든 에포크 후에 교체됩니다.

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

더 많은 예를 보려면 여기를 확인 하십시오 .


2
사이트에 오신 것을 환영합니다! 그리고 덕분에, 내가 대답에 그 또한 감사
I_Play_With_Data
당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.