python - 继续用 keras 中保存的模型训练 CNN

Question

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

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)

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

标签： pythondeep-learningkerasconvolutional-neural-network

Answer 1

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