Esempio di callback di Keras per salvare un modello dopo ogni epoca?


17

Qualcuno può pubblicare un semplice esempio di Keras usando un callback per salvare un modello dopo ogni epoca? Posso trovare esempi di risparmio di pesi, ma voglio essere in grado di salvare un modello completamente funzionante dopo ogni epoca di allenamento.

Risposte:


22

Impostando 'save_weights_only' su False nel callback di Keras 'ModelCheckpoint' si salverà il modello completo; questo esempio preso dal link sopra salverà un modello completo ogni epoca, indipendentemente dalle prestazioni:

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

Alcuni altri esempi sono disponibili qui , tra cui il salvataggio solo dei modelli migliorati e il caricamento dei modelli salvati.


10

Assicurati di includere la variabile di epoca nel percorso del file. In caso contrario, il modello salvato verrà sostituito dopo ogni epoca.

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

Per altri esempi, controlla qui .


2
Benvenuti nel sito! E grazie, apprezzo
quell'aggiunta
Utilizzando il nostro sito, riconosci di aver letto e compreso le nostre Informativa sui cookie e Informativa sulla privacy.
Licensed under cc by-sa 3.0 with attribution required.