Как сохранить свою модель keras после каждой эпохи? - PullRequest
0 голосов
/ 08 января 2020

У меня есть рабочая модель сегментации изображения, которую я использую. Я настраивал параметры model.fit(), чтобы получить лучшие результаты. В настоящее время он работает около 1000 epochs. Я хотел знать, есть ли способ включить model.save() после каждой nth эпохи, а не просто сохранить модель после последней эпохи? Я хочу сделать это так, чтобы у меня был прогноз, показывающий, как повышается точность моделей, и выполнялась короткая анимация, показывающая сегментацию изображения с течением времени.

1 Ответ

0 голосов
/ 08 января 2020

Решено:

ModelCheckpoint('models\\unet_seg_{epoch:04d}.model', monitor=observe_var, save_best_only=False, period = 2)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...