浅谈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连接池实现示例程序
Nov 26 Python
基于python的Tkinter实现一个简易计算器
Dec 31 Python
利用Python爬虫给孩子起个好名字
Feb 14 Python
详解python里使用正则表达式的全匹配功能
Oct 19 Python
ubuntu环境下python虚拟环境的安装过程
Jan 07 Python
VScode编写第一个Python程序HelloWorld步骤
Apr 06 Python
pandas.DataFrame选取/排除特定行的方法
Jul 03 Python
Python facenet进行人脸识别测试过程解析
Aug 16 Python
在python Numpy中求向量和矩阵的范数实例
Aug 26 Python
Python使用random模块生成随机数操作实例详解
Sep 17 Python
Python高级编程之继承问题详解(super与mro)
Nov 19 Python
python 在threading中如何处理主进程和子线程的关系
Apr 25 Python
tensorflow中的数据类型dtype用法说明
May 26 #Python
详解Python魔法方法之描述符类
May 26 #Python
使用tensorflow 实现反向传播求导
python基础学习之递归函数知识总结
TensorFlow的自动求导原理分析
Django分页器的用法你都了解吗
May 26 #Python
tensorflow中的梯度求解及梯度裁剪操作
May 26 #Python
You might like
扩展你的 PHP 之入门篇
2006/12/04 PHP
php函数之子字符串替换 str_replace
2011/03/23 PHP
php截取指定2个字符之间字符串的方法
2015/04/15 PHP
[原创]php求圆周率的简单实现方法
2016/05/30 PHP
Yii2使用$this->context获取当前的Module、Controller(控制器)、Action等
2017/03/29 PHP
PHP获取真实IP及IP模拟方法解析
2020/11/24 PHP
优化javascript的执行速度
2010/01/23 Javascript
JS控制阿拉伯数字转为中文大写示例代码
2013/09/04 Javascript
jQuery 和 CSS 的文本特效插件集锦
2014/12/12 Javascript
原生js页面滚动延迟加载图片
2015/12/20 Javascript
详解JavaScript 中的 replace 方法
2016/01/01 Javascript
提升jQuery的性能需要做好七件事
2016/01/11 Javascript
JavaScript-html标题滚动效果的简单实现
2016/09/08 Javascript
vuejs响应用户事件(如点击事件)
2017/03/14 Javascript
浅谈Node.js CVE-2017-14849 漏洞分析(详细步骤)
2017/11/10 Javascript
vue ssr 指南详读
2018/06/29 Javascript
简单了解vue.js数组的常用操作
2019/06/17 Javascript
JS查找孩子节点简单示例
2019/07/25 Javascript
微信小程序sessionid不一致问题解决
2019/08/30 Javascript
基于vue.js仿淘宝收货地址并设置默认地址的案例分析
2020/08/20 Javascript
python中类变量与成员变量的使用注意点总结
2017/04/29 Python
Python实现利用最大公约数求三个正整数的最小公倍数示例
2017/09/30 Python
Python编程pygame模块实现移动的小车示例代码
2018/01/03 Python
Python判断一个三位数是否为水仙花数的示例
2018/11/13 Python
Python实现的爬取小说爬虫功能示例
2019/03/30 Python
python中使用while循环的实例
2019/08/05 Python
树莓派3 搭建 django 服务器的实例
2019/08/29 Python
html5+css如何实现中间大两头小的轮播效果
2018/12/06 HTML / CSS
荷兰领先的百货商店:De Bijenkorf
2018/10/17 全球购物
CheapTickets泰国:廉价航班,查看促销价格并预订机票
2019/12/28 全球购物
学校司机岗位职责
2013/11/14 职场文书
商场消防安全责任书
2014/07/29 职场文书
业绩倒数第一的检讨书
2014/09/24 职场文书
2015年社区统计工作总结
2015/04/21 职场文书
使用Redis实现分布式锁的方法
2022/06/16 Redis
Spring Boot实现文件上传下载
2022/08/14 Java/Android