Keras回调示例,用于在每个时期后保存模型?


17

有人可以在每个时期后使用回调函数发布一个简单的Keras示例来保存模型吗?我可以找到节省权重的示例,但是我希望能够在每次训练之后都保存一个功能完整的模型。

Answers:


22

在Keras 回调 'ModelCheckpoint'中将'save_weights_only'设置为False 将保存完整模型。从上面的链接获取的此示例将在每个时期保存完整的模型,而不考虑性能如何:

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

这里可以找到更多示例,包括仅保存改进的模型和加载保存的模型。


10

确保在文件路径中包含epoch变量。否则,每个时期后,您保存的模型将被替换。

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

有关更多示例,请在此处检查。


2
欢迎光临本站!谢谢,我感谢回答的
另一方
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.