保存模型与保存模型权重

被刻印的时光 ゝ 提交于 2020-02-25 20:03:29

保存模型

将整个模型保存下来,以后直接载入模型与训练数据即可开始训练,不用再定义网络和编译模型.(这种方法已经保存了模型的结构和权重,以及损失函数和优化器)

model.save('**.h5')

载入模型

from keras.models import load_model
model.load_model('**.h5')

保存模型权重

只保存模型的权重,可以看作是保存模型的一部分.

model.save_weights(’**.h5’)

载入模型权重

model.load_weights('**.h5')

 

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!