浅谈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基础教程之类class定义使用方法
Feb 20 Python
详解Python中的装饰器、闭包和functools的教程
Apr 02 Python
python 远程统计文件代码分享
May 14 Python
将Django框架和遗留的Web应用集成的方法
Jul 24 Python
实例解析Python设计模式编程之桥接模式的运用
Mar 02 Python
NumPy 数学函数及代数运算的实现代码
Jul 18 Python
Python实现多属性排序的方法
Dec 05 Python
基于Python+Appium实现京东双十一自动领金币功能
Oct 31 Python
基于jupyter代码无法在pycharm中运行的解决方法
Apr 21 Python
8种常用的Python工具
Aug 05 Python
Python读取Excel一列并计算所有对象出现次数的方法
Sep 04 Python
使用numpngw和matplotlib生成png动画的示例代码
Jan 24 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 Http_Template_IT类库进行模板替换
2009/03/19 PHP
深入理解PHP中mt_rand()随机数的安全
2017/10/12 PHP
php传值和传引用的区别点总结
2019/11/19 PHP
jQuery get和post 方法传值注意事项
2009/11/03 Javascript
使用upstart把nodejs应用封装为系统服务实例
2014/06/01 NodeJs
JavaScript中遍历对象的property的3种方法介绍
2014/12/30 Javascript
原生js和jquery实现图片轮播特效
2015/04/23 Javascript
window.open打开窗口被拦截的快速解决方法
2016/08/04 Javascript
Vue.js实现文章评论和回复评论功能
2020/05/30 Javascript
React+Webpack快速上手指南(小结)
2018/08/15 Javascript
jQuery实现基本动画效果的方法详解
2018/09/06 jQuery
微信小程序ibeacon三点定位详解
2018/10/31 Javascript
原生js实现Flappy Bird小游戏
2018/12/24 Javascript
在layui中select更改后生效的方法
2019/09/05 Javascript
微信小程序实现选择地址省市区三级联动
2020/06/21 Javascript
jQuery实现评论模块
2020/08/19 jQuery
Python编程中的for循环语句学习教程
2015/10/14 Python
Java多线程编程中ThreadLocal类的用法及深入
2016/06/21 Python
python 编程之twisted详解及简单实例
2017/01/28 Python
Python实现备份MySQL数据库的方法示例
2018/01/11 Python
使用pandas读取csv文件的指定列方法
2018/04/21 Python
pyhton中__pycache__文件夹的产生与作用详解
2019/11/24 Python
Python+Redis实现布隆过滤器
2019/12/08 Python
Django自定义全局403、404、500错误页面的示例代码
2020/03/08 Python
如何基于Python爬取隐秘的角落评论
2020/07/02 Python
基于django和dropzone.js实现上传文件
2020/11/24 Python
python给list排序的简单方法
2020/12/10 Python
CSS超出文本指定宽度用省略号代替和文本不换行
2016/05/05 HTML / CSS
乐高奥地利官方商店:LEGO Shop AT
2019/07/16 全球购物
党的群众路线对照检查材料
2014/08/27 职场文书
社保代办委托书怎么写
2014/10/06 职场文书
2016年大学生党员公开承诺书
2016/03/24 职场文书
CSS filter 有什么神奇用途
2021/05/25 HTML / CSS
IDEA使用SpringAssistant插件创建SpringCloud项目
2021/06/23 Java/Android
利用Python读取微信朋友圈的多种方法总结
2021/08/23 Python
vue3不同环境下实现配置代理
2022/05/25 Vue.js