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

时间:2018-05-01 19:33:08

标签: python tensorflow keras tensorboard

使用Keras模型,我已经包含TensorBoard回调以生成日志文件,以便稍后可视化。

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

这导致了TensorBoard中的不可开发的图表(下面的截图)。

使用原始Tensorflow,我已经看到这可以通过添加" global_step"来解决。张量来跟踪运行之间的纪元数。

但我怎么能用Keras做到这一点?

Glitchy graphs in Tensorboard

1 个答案:

答案 0 :(得分:2)

model.fit默认使用参数initial_epoch,0,可让您告诉模型epoch它的起始位置。您可以使用它来恢复以前的培训。