Exemplo de Keras Callback para salvar um modelo após cada época?

17

Alguém pode postar um exemplo direto de Keras usando um retorno de chamada para salvar um modelo após cada época? Posso encontrar exemplos de economia de pesos, mas quero poder salvar um modelo completamente funcional após cada época de treinamento.

I_Play_With_Data
fonte

Respostas:

22

Definir 'save_weights_only' como False no retorno de chamada Keras 'ModelCheckpoint' salvará o modelo completo; este exemplo, retirado do link acima, salvará um modelo completo a cada época, independentemente do desempenho:

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

Mais alguns exemplos são encontrados aqui , incluindo salvar apenas modelos aprimorados e carregar os modelos salvos.

redhqs
fonte
10

Certifique-se de incluir a variável epoch em seu caminho de arquivo. Caso contrário, seu modelo salvo será substituído após cada época.

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

Para mais exemplos, verifique aqui .

Safwan
fonte
2
Bem vindo ao site! E obrigado, agradeço essa adição à resposta
I_Play_With_Data 28/02/19