Python 继续使用keras中保存的模型训练CNN
我用keras库训练CNN模型,历元数为25。我是否可以第一次使用10个纪元运行模型,然后使用以下代码行保存模型:Python 继续使用keras中保存的模型训练CNN,python,deep-learning,keras,convolutional-neural-network,Python,Deep Learning,Keras,Convolutional Neural Network,我用keras库训练CNN模型,历元数为25。我是否可以第一次使用10个纪元运行模型,然后使用以下代码行保存模型: model.fit_generator(training_set, steps_per_epoch = 100000, epochs = 10, validation_data = test_set, validat
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个时代,如下面的代码所示:
model = load_model('my_model.h5')
model.fit_generator(training_set,
steps_per_epoch = 100000,
epochs = 15,
validation_data = test_set,
validation_steps = 40000)
继续培训是否足够?或者我必须做任何其他步骤来继续工作。非常感谢您的支持。是的,这没关系,
model.save
保存权重、模型体系结构和优化器状态,这样您就可以毫无问题地恢复培训。谢谢,我担心重新启动python后,模型将从一开始就开始培训。我真的很感谢你的支持。