Keras Callback-Beispiel zum Speichern eines Modells nach jeder Epoche?


16

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:


20

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.


7

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 .


2
Willkommen auf der Seite! Und danke, ich schätze diesen Zusatz zur Antwort
I_Play_With_Data
Durch die Nutzung unserer Website bestätigen Sie, dass Sie unsere Cookie-Richtlinie und Datenschutzrichtlinie gelesen und verstanden haben.
Licensed under cc by-sa 3.0 with attribution required.