Как сгенерировать уникальные имена для сохраненной модели каждой эпохи с помощью API-интерфейса Keras от TensorFlow - PullRequest
0 голосов
/ 11 апреля 2019

Я тренируюсь с fit_generator() и хочу создать уникальные имена для сохраненных способностей каждой эпохи

Уже пробовал: см. Код позади

Код:

model_path = '.\checkpoints\cp{}.ckpt'.format(time())
cp_callback = tf.keras.callbacks.ModelCheckpoint(model_path, 
                                                 verbose=1,
                                                  period=2)
model.fit_generator(..........,callbacks=[cp_callback])

Ожидается: Для создания уникального имени контрольной точки
например epoch_4.ckpt или epoch_5.ckpt
Факт: Каждый раз, когда он сохраняет, перезаписывает существующую контрольную точку

1 Ответ

1 голос
/ 11 апреля 2019

Вы можете попробовать добавить переменную epoch к переменной filepath

filepath = ".\checkpoints\cp-{epoch:02d}.hdf5"
cp_callback = tf.keras.callbacks.ModelCheckpoint(filepath, verbose=1, period=2)

В документации Keras здесь уже упоминается это.

...