栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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

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】训练模型以及保存最佳模型

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/461665.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号