keras中保存与调用预训练(离线)模型

      

目录

保存模型的全部配置信息

加载预训练模型


  在遇到数据量庞大的数据集时,训练时长是一个巨大的问题。有时已经完成了模型参数调优的步骤,但若没保存模型,每次实验都得重新训练,才能进行下一步的调整,这大大增加了时间与精力。此时保存模型的h5模型就尤为重要了。

保存模型的全部配置信息

model.save('LSTM.h5')

加载预训练模型

from keras.models import load_model
model = load_model('LSTM.h5')

加载完模型就能得到与之前保存的一模一样的模型与模型参数:

keras中保存与调用预训练(离线)模型_第1张图片

 

你可能感兴趣的:(深度学习,keras,深度学习)