深度学习 走进tensorflow2.0】tensorflow2.0 保存和加载模型

这篇博客详细介绍了在TensorFlow2.0中如何保存和加载模型,包括使用H5格式保存全模型以及采用SavedModel格式进行保存和加载。教程内容丰富,适合初学者,通过实例展示了模型保存的两种主要方法。
摘要由CSDN通过智能技术生成

无意中发现了一个巨牛的人工智能教程,忍不住分享一下给大家。教程不仅是零基础,通俗易懂,而且非常风趣幽默,像看小说一样!觉得太牛了,所以分享给大家。点这里可以跳转到教程。人工智能教程

tensorflow2.0 保存模型方式主要有4种,下面一一介绍下。

1、保存全模型,权重和模型框架一起保存H5格式

model.save('the_save_model.h5')
new_model = keras.models.load_model('the_save_model.h5')

2、将模型保存为SavedModel格式,python,java 均可调用

keras.experimental.export_saved_model(model, 'saved_model')
new_model = keras.experimental.load_from_saved_model('saved_model')
或者
# 保存整个模型,保存成SavedModel格式
model.save('serving_model2/', save_format='tf')
# 保存整个模型成SavedModel格式,注意参数'my_saved_model'是文件夹名称,不是文件哦!
tf.saved_model.save(model, 'serving_model/')
# 从SavedModel加载模型
new_model2 = load_model('serving_model/')
print(new_model2)


在这里插入图片描述

3、其他方式参考我上一篇博客。
【深度学习 走进tensorflow2.0】训练的模型保存方式
【深度学习 走进tensorflow2.0】训练模型以及保存最佳模型

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

东华果汁哥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值