Quelqu'un peut-il poster un exemple simple de Keras utilisant un rappel pour enregistrer un modèle après chaque époque? Je peux trouver des exemples d'économies de poids, mais je veux pouvoir enregistrer un modèle complètement fonctionnel après chaque époque d'entraînement.
La définition de «save_weights_only» sur False dans le rappel Keras «ModelCheckpoint» enregistrera le modèle complet; cet exemple tiré du lien ci-dessus enregistrera un modèle complet à chaque époque, quelles que soient les performances:
keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)Vous trouverez ici d' autres exemples , notamment l'enregistrement de modèles améliorés uniquement et le chargement des modèles enregistrés.
Assurez-vous d'inclure la variable epoch dans votre chemin de fichier. Sinon, votre modèle enregistré sera remplacé après chaque époque.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')
Pour plus d'exemples, consultez ici .