Как сохранить номер эпохи с Keras при выполнении нескольких запусков?

Question

В модели Keras я включил обратный вызов TensorBoard, чтобы сгенерировать файлы журналов для последующей визуализации.

Проблема в том, что если я тренирую свою модель несколько раз, она генерирует несколько файлов журналов, и номер шага всегда перезапускается до 0, а не продолжается на последнем шаге предыдущего запуска.

Это приводит к неэксплуатационному графу в TensorBoard (скриншот ниже).

С необработанным Tensorflow я видел, что это можно решить, добавив тензор "global_step", чтобы отслеживать число эпох между циклами.

Но как мне это сделать с помощью Keras?

Answer 1

model.fit имеет аргумент initial_epoch, по умолчанию 0, который позволяет указать модели, с которой epoch начинается. Вы можете использовать это, чтобы возобновить предыдущее обучение.