Может кто-нибудь опубликовать простой пример использования Keras обратного вызова для сохранения модели после каждой эпохи? Я могу найти примеры экономии веса, но я хочу иметь возможность сохранять полностью функционирующую модель после каждой тренировочной эпохи.
Ответы:
Установка «save_weights_only» в False в Keras обратного вызова «ModelCheckpoint» будет сохранить полную модель; этот пример, взятый по ссылке выше, будет сохранять полную модель каждую эпоху, независимо от производительности:
keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)Некоторые другие примеры можно найти здесь , в том числе сохранение только улучшенные модели и загрузки сохраненных моделей.
Не забудьте включить переменную epoch в путь к файлу. В противном случае ваша сохраненная модель будет заменена после каждой эпохи.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')Дополнительные примеры, проверить здесь .