keras模型保存和加载

(一)保存和加载整个模型

​ 包含模型的结构、权重、训练配置项(损失函数、优化器)、优化器状态,允许准确地从上次结束的地方开始训练。

1.训练完模型后
path='.../.../xxx.h5'
model.save(path)
2.调用模型
#1 调用模型
import keras
from keras.models import load_model
path='.../.../xxx.h5'
my_model=load_model(path)
#2 引入并处理数据(因数据而异)
#3 可以直接开始evaluate/predict/继续训练
(二)只保存、加载模型结构

保存模型的结构,而非其权重或训练配置项(未经compile)。

1.训练完模型后
my_model_json=model.to_json()
path='.../.../xxx.json'
with open(path,'w') as f:
    f.write(my_model_json)
2.调用模型
import keras
from keras.models import model_from_json
path='.../.../xxx.json'
with open(path,'r') as f:
    my_model_json=f.read()
my_json_model=model_from_json(my_model_json)
my_json_model.summary()
(三)只保存/加载模型的权重
1.训练完模型后
path='.../.../xx.h5'
model.save_weights(path)
2.加载
①非按名
path='.../.../xx.h5'
model.load_weights(path)
②按名(可用来只加载部分权重——只需new_model与原model的名称部分相同部分不同即可)
path='.../.../xx.h5'
model.load_weights(path,by_name=True)
(四)checkpoint
1.训练模型
checkpoint_path='.../.../xx'
cp_callback=keras.callback.ModelCheckpoint(checkpoint_path, monitor='val_loss',verbose=0, save_best_only=False,save_weights_only=False,mode='auto', save_freq='epoch', options=None)
history=model.fit(train_image,train_label,epochs=3,batch_size=512,validation_data=(test_image,test_label),callbacks=[cp_callback])
2.调用

同上1/3中的调用,只需把path换成checkpoint_path

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值