python - Keras 在额外的历元继续训练后正确保存检查点 - 初始历元

Question

当我决定训练新模型并保存我希望保存的检查点时，

ModelCheckpoint 效果很好。然而，当我决定将相同的模型训练n多个时期时，我的问题就出现了。问题是纪元重置为 0，这将产生一些模型检查点名称，如下所示:

/checkpoints
    checkpoint-01-0.24.h5
    checkpoint-02-0.34.h5
    checkpoint-03-0.37.h5
              .
              .
    checkpoint-m-0.68.h5
    checkpoint-01-0.71.h5
    checkpoint-02-0.73.h5
    checkpoint-03-0.74.h5
              .
              .
    checkpoint-n-0.85.h5

如您所见，纪元将被重置。我想要实现的是获取先前迭代中所有纪元的数量并添加新纪元以获得如下所示的内容:

    checkpoint-(m + 01)-0.71.h5
    checkpoint-(m + 02)-0.73.h5
    checkpoint-(m + 03)-0.74.h5
              .
              .
    checkpoint-(m + n)-0.85.h5

Answer 1

正如您可以在 the doc 中读到的那样在 .fit() 函数中，有一个参数正是执行此操作:

initial_epoch: epoch at which to start training (useful for resuming a previous training run)

所以只需添加:

model.fit(..., initial_epoch=m)

如您的示例中所示，m 是要运行的第一个纪元。

我希望这有帮助:)