¿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.

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.


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í .


2
Bienvenido al sitio! Y gracias, agradezco esa adición a la respuesta
I_Play_With_Data
Al usar nuestro sitio, usted reconoce que ha leído y comprende nuestra Política de Cookies y Política de Privacidad.
Licensed under cc by-sa 3.0 with attribution required.