博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
tensorflow初探---模型文件保存和加载
阅读量:7100 次
发布时间:2019-06-28

本文共 607 字,大约阅读时间需要 2 分钟。

hot3.png

tensorflow模型文件可以大致分为两种:

1.通过tensorflow.train.Saver来生成

1)保存

saver = tf.train.Saver(tf.all_variables())saver.save(sess, 'model.ckpt')

2)加载

saver=tf.train.Saver(tf.all_variables())save.restore(sess, 'model.ckpt')

2.通过tensorflow.GraphDef

1)保存

tf.train.write_graph(sess.graph_def, path, filename)或者witf tf.gfile.FastGFile(filename, 'wb') as f:    f.write(sess.graph_def.SerializeToString())

2)加载

with tf.gfile.FastGFile(filename, 'rb') as f:    graph_def = tf.GraphDef()    graph_def.ParseFromString(f.read())    tf.import_graph_def(graph_def, name='')

暂时写这些,未完待续

转载于:https://my.oschina.net/u/1384571/blog/808826

你可能感兴趣的文章
博文共赏:Node.js静态文件服务器实战
查看>>
CS安装卸载测试总结(转)
查看>>
深入理解JavaScript系列(18):面向对象编程之ECMAScript实现(推荐)
查看>>
iphone开发之轻松搞定原生socket 编程,阻塞与非阻塞,收发自如
查看>>
ColdFusion select option 用法,看看哪种适合你的
查看>>
Amazium - 响应式 CSS 框架 - 开源中国
查看>>
使用Vitamio打造自己的Android万能播放器(5)——在线播放(播放优酷视频)
查看>>
iis7 发布mvc 遇到的HTTP错误 403.14-Forbidden Web 服务器被配置为不列出此目录的内容...
查看>>
PHP通过Thrift操作Hbase
查看>>
Sql Server导入Access数据库报不可识别的数据库格式 Microsoft JET Database Engine
查看>>
存储设备形成的层次结构
查看>>
http://knockoutjs.com/工作杂记
查看>>
Http协议中的Header与Body
查看>>
Android项目环境搭建
查看>>
窗口大小改变绑定resize事件
查看>>
python数据结构之二叉树遍历的实现
查看>>
进出口流程 & 报关单据
查看>>
各主流浏览器内核介绍
查看>>
[LeetCode] Copy List with Random Pointe
查看>>
我使用中的Linux命令和快捷键(For Ubuntu)
查看>>