Пример Keras Callback для сохранения модели после каждой эпохи?

17

Может кто-нибудь опубликовать простой пример использования Keras обратного вызова для сохранения модели после каждой эпохи? Я могу найти примеры экономии веса, но я хочу иметь возможность сохранять полностью функционирующую модель после каждой тренировочной эпохи.

I_Play_With_Data
источник

Ответы:

22

Установка «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)

Некоторые другие примеры можно найти здесь , в том числе сохранение только улучшенные модели и загрузки сохраненных моделей.

redhqs
источник
10

Не забудьте включить переменную epoch в путь к файлу. В противном случае ваша сохраненная модель будет заменена после каждой эпохи.

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

Дополнительные примеры, проверить здесь .

Сафван
источник
2
Добро пожаловать на сайт! И спасибо, я ценю это дополнение к ответу
I_Play_With_Data