浅谈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中的floor()方法
May 15 Python
Python 判断 有向图 是否有环的实例讲解
Feb 01 Python
python使用Matplotlib绘制分段函数
Sep 25 Python
python 字符串常用方法汇总详解
Sep 16 Python
DJango的创建和使用详解(默认数据库sqlite3)
Nov 18 Python
使用Python生成200个激活码的实现方法
Nov 22 Python
pandas分批读取大数据集教程
Jun 06 Python
python 实现rolling和apply函数的向下取值操作
Jun 08 Python
Python Scrapy图片爬取原理及代码实例
Jun 12 Python
如何基于Python代码实现高精度免费OCR工具
Jun 18 Python
python删除文件、清空目录的实现方法
Sep 23 Python
Python语言中的数据类型-序列
Feb 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 Smarty生成EXCEL文档的代码
2008/08/23 PHP
ThinkPHP分页实例
2014/10/15 PHP
Yii入门教程之目录结构、入口文件及路由设置
2014/11/25 PHP
thinkphp如何获取客户端IP
2015/11/03 PHP
PHP生成可点击刷新的验证码简单示例
2016/05/13 PHP
解析javascript系统错误:-1072896658的解决办法
2013/07/08 Javascript
js 距离某一时间点时间是多少实现代码
2013/10/14 Javascript
jQuery控制cookie过期时间的方法
2015/04/07 Javascript
实例详解Nodejs 保存 payload 发送过来的文件
2016/01/14 NodeJs
jQuery插件实现带圆点的焦点图片轮播切换
2016/01/18 Javascript
jQuery绑定事件监听bind和移除事件监听unbind用法实例详解
2016/01/19 Javascript
jQuery+css实现的换页标签栏效果
2016/01/27 Javascript
分类解析jQuery选择器
2016/11/23 Javascript
es7学习教程之Decorators(修饰器)详解
2017/07/21 Javascript
JavaScript之事件委托实例(附原生js和jQuery代码)
2017/07/22 jQuery
写gulp遇到的ES6问题详解
2018/12/03 Javascript
javascript 设计模式之享元模式原理与应用详解
2020/04/08 Javascript
深入浅析vue全局环境变量和模式
2020/04/28 Javascript
Python中内置的日志模块logging用法详解
2016/07/12 Python
Python Print实现在输出中插入变量的例子
2019/12/25 Python
零基础学python应该从哪里入手
2020/08/11 Python
Python可以用来做什么
2020/11/23 Python
linux centos 7.x 安装 python3.x 替换 python2.x的过程解析
2020/12/14 Python
详解css3 flex弹性盒自动铺满写法
2020/09/17 HTML / CSS
深入研究HTML5实现图片压缩上传功能
2016/03/25 HTML / CSS
家用个人磨皮机:Trophy Skin
2017/03/30 全球购物
英国办公家具网站:Furniture At Work
2019/10/07 全球购物
销售会计岗位职责
2014/03/15 职场文书
员工保密承诺书
2014/05/28 职场文书
婚前协议书范本两则
2014/10/16 职场文书
2014年学校食堂工作总结
2014/11/25 职场文书
离婚协议书怎么写
2015/01/26 职场文书
物业工程部岗位职责
2015/02/11 职场文书
学校重阳节活动总结
2015/03/24 职场文书
研究生论文答辩开场白
2015/05/27 职场文书
校园开放日新闻稿
2015/07/17 职场文书