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

Posted 2023-03-27

【中文标题】执行多次运行时如何使用 Keras 保留纪元数？

【英文标题】：How to preserve the epoch number with Keras when performing multiple runs?

【发布时间】：2018-10-11 20:32:23

【问题描述】：

在 Keras 模型中，我包含了 TensorBoard 回调来生成日志文件以供稍后可视化。

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

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

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

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

【参考方案1】：

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