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.
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.
Respostas:
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.
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 .