Чи може хтось, будь ласка, опублікувати простий приклад 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)Деякі інші приклади можна знайти тут , в тому числі збереження тільки поліпшені моделі і завантаження збережених моделей.
Не забудьте включити змінну епохи у свій шлях. Інакше збережена модель буде замінена після кожної епохи.
filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')
Щоб отримати додаткові приклади, перегляньте тут .