¿Ejemplo de devolución de llamada de Keras para guardar un modelo después de cada época?

17

¿Alguien puede publicar un ejemplo directo de Keras usando una devolución de llamada para guardar un modelo después de cada época? Puedo encontrar ejemplos de ahorro de pesas, pero quiero poder guardar un modelo completamente funcional después de cada época de entrenamiento.

I_Play_With_Data
fuente

Respuestas:

22

Establecer 'save_weights_only' en False en la devolución de llamada de Keras 'ModelCheckpoint' guardará el modelo completo; Este ejemplo tomado del enlace anterior guardará un modelo completo en cada época, independientemente del rendimiento:

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

Aquí se encuentran algunos ejemplos más , que incluyen guardar solo modelos mejorados y cargar los modelos guardados.

redhqs
fuente
10

Asegúrese de incluir la variable epoch en su ruta de archivo. De lo contrario, su modelo guardado será reemplazado después de 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 más ejemplos, marque aquí .

Safwan
fuente
2
Bienvenido al sitio! Y gracias, agradezco esa adición a la respuesta
I_Play_With_Data