执行多次运行时如何使用 Keras 保留纪元号？

Question

对于 Keras 模型，我添加了 TensorBoard 回调来生成稍后可视化的日志文件。

问题是，如果我多次训练模型，它会生成多个日志文件，并且步骤号总是重新启动为 0，而不是继续上次运行的最后一步。

这会导致 TensorBoard 中的图表无法利用（下面的屏幕截图）。

使用原始 Tensorflow，我发现可以通过添加“global_step”张量来跟踪运行之间的纪元数来解决这个问题。

但是我如何使用 Keras 来做到这一点？

Answer 1

model.fit有一个参数initial_epoch（默认为 0），它可以让您告诉模型epoch它从哪个位置开始。您可以使用它来恢复之前的训练。