浅谈tf.train.Saver()与tf.train.import_meta_graph的要点


Posted in Python onMay 26, 2021

(一)、tf.train.Saver()

(1). tf.train.Saver() 是用来保存tensorflow训练模型的,默认保存全部参数

(2). 用来加载参数,注:只加载存储在data中的权重和偏置项等需要训练的参数,其他一律不加载,包括meta文件中的图也不加载

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

(二)、tf.train.import_meta_graph

(1). 用来加载meta文件中的图,以及图上定义的结点参数包括权重偏置项等需要训练的参数,也包括训练过程生成的中间参数,所有参数都是通过graph调用接口get_tensor_by_name(name="训练时的参数名称")来获取

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

(三)、总结

(1). 保存使用tf.train.Saver()

(2). 加载可以使用tf.train.import_meta_graph(".meta文件"),直接通过训练参数名称就可以获取需要的参数,但需要提前知道训练时的参数名称才能获取,要懂得tensorflow命名规则

(3). tf.train.Saver("./checkpoints目录/")加载的缺点是只加载了训练参数,并且必须定义与之相同(shape,dtype要相同,tf.type要相同,如:我是placeholder,你也是tf.placeholder)方能使用,当你要获取训练中间参数时,需要和训练过程一样搭建相同的网络.

补充:tf.train.import_meta_graph报KeyError

我在模型恢复时,在执行tf.train.import_meta_graph的时候报错

浅谈tf.train.Saver()与tf.train.import_meta_graph的要点

后来发现,我的模型是在服务器上训练的,服务器上tensorflow版本是1.11.0,而我在本地电脑上执行的 tf.train.import_meta_graph,我本地的tensorflow是1.5.0,我将tensorflow更新到1.11.0后,就解决了。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
深入理解Python中的元类(metaclass)
Feb 14 Python
Python中的左斜杠、右斜杠(正斜杠和反斜杠)
Aug 30 Python
Python编程对列表中字典元素进行排序的方法详解
May 26 Python
Python实现可设置持续运行时间、线程数及时间间隔的多线程异步post请求功能
Jan 11 Python
使用python脚本实现查询火车票工具
Jul 19 Python
python2.7和NLTK安装详细教程
Sep 19 Python
Python WEB应用部署的实现方法
Jan 02 Python
python tkinter库实现气泡屏保和锁屏
Jul 29 Python
Python图像读写方法对比
Nov 16 Python
scrapy头部修改的方法详解
Dec 06 Python
Python实现Appium端口检测与释放的实现
Dec 31 Python
Django利用AJAX技术实现博文实时搜索
May 06 Python
tensorflow中的数据类型dtype用法说明
May 26 #Python
详解Python魔法方法之描述符类
May 26 #Python
使用tensorflow 实现反向传播求导
python基础学习之递归函数知识总结
TensorFlow的自动求导原理分析
Django分页器的用法你都了解吗
May 26 #Python
tensorflow中的梯度求解及梯度裁剪操作
May 26 #Python
You might like
桌面中心(三)修改数据库
2006/10/09 PHP
说明的比较细的php 正则学习实例
2008/07/30 PHP
php 应用程序安全防范技术研究
2009/09/25 PHP
解析:通过php socket并借助telnet实现简单的聊天程序
2013/06/18 PHP
使用PHP和HTML5 FormData实现无刷新文件上传教程
2014/09/06 PHP
Symfony学习十分钟入门经典教程
2016/02/03 PHP
PHP array_shift()用法实例分析
2019/01/07 PHP
一个cssQuery对象 javascript脚本实现代码
2009/07/21 Javascript
IE8 下的Js错误HTML Parsing Error...
2009/08/14 Javascript
心扬JS分页函数代码
2010/09/10 Javascript
js函数的延迟加载实现代码
2012/10/11 Javascript
可自定义速度的js图片无缝滚动示例分享
2014/01/20 Javascript
jQuery实现手机自定义弹出输入框
2016/06/13 Javascript
非常优秀的JS图片轮播插件Swiper的用法
2017/01/03 Javascript
vue移动端html5页面根据屏幕适配的四种解决方法
2018/10/19 Javascript
vue实现在线翻译功能
2019/09/27 Javascript
Python数字图像处理之霍夫线变换实现详解
2018/01/12 Python
python方向键控制上下左右代码
2018/01/20 Python
transform python环境快速配置方法
2018/09/27 Python
python 解压pkl文件的方法
2018/10/25 Python
Laravel+Dingo/Api 自定义响应的实现
2019/02/17 Python
Python循环结构的应用场景详解
2019/07/11 Python
详解CSS3的opacity属性设置透明效果的用法
2016/05/09 HTML / CSS
CSS3模拟IOS滑动开关效果
2016/09/28 HTML / CSS
中国跨境电子商务网站:NewFrog
2018/03/10 全球购物
意大利火车票和铁路通行证专家:ItaliaRail
2019/01/22 全球购物
Clarks西班牙官方在线商店:clarks鞋
2019/05/03 全球购物
个人能力自我鉴赏
2014/01/25 职场文书
幼儿园大班评语大全
2014/04/17 职场文书
财务部绩效考核方案
2014/05/04 职场文书
2014年人大工作总结
2014/12/10 职场文书
旗帜观后感
2015/06/08 职场文书
初中英语教学反思范文
2016/02/15 职场文书
Python与C++中梯度方向直方图的实现
2022/03/17 Python
Java 垃圾回收超详细讲解记忆集和卡表
2022/04/08 Java/Android
Mybatis-Plus 使用 @TableField 自动填充日期
2022/04/26 Java/Android