python - 如何在 Keras 的每个时期保存训练历史？

Question

我不能让我的电脑整天都在运行，为此我需要在每个 epoch 之后保存训练历史。例如，我在一天内训练了我的模型 100 个 epoch，第二天，我想再训练它 50 个 epoch。我需要为整个 150 个时期生成损失与时期和准确性与时期的图表。我正在使用 fit_generator 方法。有没有什么方法可以在每个时期后保存训练历史(最有可能使用 Callback)？我知道如何在训练结束后保存训练历史。我正在使用 Tensorflow 后端。

Answer 1

Keras 具有 CSVLogger 回调，它似乎可以完全满足您的需求；来自 documentation :

Callback that streams epoch results to a CSV file.

它有一个用于添加到文件的附加参数。同样，来自文档:

append: Boolean. True: append if file exists (useful for continuing training). False: overwrite existing file

from keras.callbacks import CSVLogger

csv_logger = CSVLogger("model_history_log.csv", append=True)
model.fit_generator(...,callbacks=[csv_logger])