Kann jemand bitte ein einfaches Beispiel für Keras posten, indem er einen Rückruf verwendet, um ein Modell nach jeder Epoche zu speichern? Ich kann Beispiele für das Speichern von Gewichten finden, möchte aber nach jeder Trainingsepoche ein voll funktionsfähiges Modell speichern können.
Antworten:
Wenn Sie im Keras- Rückruf 'ModelCheckpoint' 'save_weights_only' auf False setzen, wird das vollständige Modell gespeichert. Dieses Beispiel aus dem obigen Link speichert in jeder Epoche ein vollständiges Modell, unabhängig von der Leistung:
keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)Einige weitere Beispiele finden sich hier , darunter nur eine verbesserte Modelle speichern und die gespeicherten Modelle laden.
Stellen Sie sicher, dass Sie die Epochenvariable in Ihren Dateipfad aufnehmen. Andernfalls wird Ihr gespeichertes Modell nach jeder Epoche ersetzt.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')Weitere Beispiele finden Sie hier .