浅谈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之模拟鼠标键盘动作具体实现
Dec 30 Python
python中使用urllib2获取http请求状态码的代码例子
Jul 07 Python
Python多线程实例教程
Sep 06 Python
python定时检查某个进程是否已经关闭的方法
May 20 Python
用Python登录好友QQ空间点赞的示例代码
Nov 04 Python
TensorFlow模型保存和提取的方法
Mar 08 Python
Pytorch Tensor基本数学运算详解
Dec 30 Python
浅谈pytorch卷积核大小的设置对全连接神经元的影响
Jan 10 Python
使用tensorflow实现矩阵分解方式
Feb 07 Python
Python编程快速上手——选择性拷贝操作案例分析
Feb 28 Python
手把手教你怎么用Python实现zip文件密码的破解
May 27 Python
python和Appium的移动端多设备自动化测试框架
Apr 26 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-MySQL教程归纳总结
2008/06/07 PHP
php setcookie(name, value, expires, path, domain, secure) 参数详解
2013/06/28 PHP
键盘上一张下一张兼容IE/google/firefox等浏览器
2014/01/28 Javascript
javascript如何判断输入的url是否正确
2014/04/11 Javascript
javascript字母大小写转换的4个函数详解
2014/05/09 Javascript
由浅入深讲解Javascript继承机制与simple-inheritance源码分析
2015/12/13 Javascript
JavaScript直播评论发弹幕切图功能点集合效果代码
2016/06/26 Javascript
Vue.js第四天学习笔记
2016/12/02 Javascript
js实现带缓动动画的导航栏效果
2017/01/16 Javascript
NodeJS实现图片上传代码(Express)
2017/06/30 NodeJs
微信小程序 函数防抖 解决重复点击消耗性能问题实现代码
2019/09/12 Javascript
[01:45]DOTA2新英雄“神谕者”全方位展示
2014/11/21 DOTA
[44:09]DOTA2上海特级锦标赛A组小组赛#1 EHOME VS MVP.Phx第二局
2016/02/25 DOTA
python访问类中docstring注释的实现方法
2015/05/04 Python
详解Python的Flask框架中生成SECRET_KEY密钥的方法
2016/06/07 Python
Zabbix实现微信报警功能
2016/10/09 Python
Python根据已知邻接矩阵绘制无向图操作示例
2018/06/23 Python
python 除法保留两位小数点的方法
2018/07/16 Python
Python 学习教程之networkx
2019/04/15 Python
python GUI实现小球满屏乱跑效果
2019/05/09 Python
Python如何实现转换URL详解
2019/07/02 Python
利用pytorch实现对CIFAR-10数据集的分类
2020/01/14 Python
英国的潮牌鞋履服饰商店:size?
2019/03/26 全球购物
美国翻新电子产品商店:The Store
2019/10/08 全球购物
定义一结构体变量,用其表示点坐标,并输入两点坐标,求两点之间的距离
2015/08/17 面试题
技校教师求职简历的自我评价
2013/10/20 职场文书
博士生入学考试推荐信
2013/11/17 职场文书
音乐节策划方案
2014/06/09 职场文书
群众路线个人整改措施
2014/10/24 职场文书
党员个人总结范文
2015/02/14 职场文书
好好学习保证书
2015/02/26 职场文书
英语通知范文
2015/04/22 职场文书
MySQL Shell的介绍以及安装
2021/04/24 MySQL
jdbc使用PreparedStatement批量插入数据的方法
2021/04/27 MySQL
Java实现多文件上传功能
2021/06/30 Java/Android
golang中字符串MD5生成方式总结
2021/07/04 Golang