继续用 keras 中保存的模型训练 CNN答案

【问题标题】：Continue train CNN with saved model in keras继续用 keras 中保存的模型训练 CNN
【发布时间】：2018-10-16 11:33:50
【问题描述】：

我用 keras 库训练 CNN 模型，epoch 数为 25。我可以第一次运行 10 个 epoch 的模型，然后用这些代码行保存模型：

model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 10,
                     validation_data = test_set,
                     validation_steps = 40000)
from keras.models import load_model
model.save('my_model.h5')

然后我重新启动 python 并继续使用相同的数据集运行接下来的 15 个 epoch，如下面的代码：

model = load_model('my_model.h5')
model.fit_generator(training_set,
                     steps_per_epoch = 100000,
                     epochs = 15,
                     validation_data = test_set,
                     validation_steps = 40000)

继续训练就够了吗？或者我必须执行任何其他步骤才能继续工作。我非常感谢任何支持。

【问题讨论】：

标签： python deep-learning keras convolutional-neural-network

【解决方案1】：

是的，这没关系，model.save 保存了权重、模型架构和优化器状态，因此您可以毫无问题地继续训练。

【讨论】：

谢谢，我害怕重启python后模型会从头开始训练。非常感谢您的支持。