누군가가 각 시대마다 모델을 저장하기 위해 콜백을 사용하여 Keras의 간단한 예를 게시 할 수 있습니까? 웨이트 절약의 예를 찾을 수 있지만 모든 훈련 시대마다 완벽하게 작동하는 모델을 저장할 수 있기를 원합니다.
누군가가 각 시대마다 모델을 저장하기 위해 콜백을 사용하여 Keras의 간단한 예를 게시 할 수 있습니까? 웨이트 절약의 예를 찾을 수 있지만 모든 훈련 시대마다 완벽하게 작동하는 모델을 저장할 수 있기를 원합니다.
답변:
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)
개선 된 모델 만 저장하고 저장된 모델을로드하는 등 몇 가지 예가 여기에 있습니다 .
파일 경로에 epoch 변수를 포함시켜야합니다. 그렇지 않으면 저장된 모델은 모든 에포크 후에 교체됩니다.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')