Tensorflow: 从checkpoint文件中读取tensor方式


Posted in Python onFebruary 10, 2020

在使用pre-train model时候,我们需要restore variables from checkpoint files.

经常出现在checkpoint 中找不到”Tensor name not found”.

这时候需要查看一下ckpt中到底有哪些变量

import os
from tensorflow.python import pywrap_tensorflow

checkpoint_path = os.path.join(model_dir, "model.ckpt")
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
var_to_shape_map = reader.get_variable_to_shape_map()
# Print tensor name and values
for key in var_to_shape_map:
 print("tensor_name: ", key)
 print(reader.get_tensor(key))

可以显示ckpt中的tensor名字和值,当然也可以用pycharm调试。

以上这篇Tensorflow: 从checkpoint文件中读取tensor方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python生成IP段的方法
Jul 07 Python
python开发之tkinter实现图形随鼠标移动的方法
Nov 11 Python
Python使用time模块实现指定时间触发器示例
May 18 Python
python+selenium+autoit实现文件上传功能
Aug 23 Python
python Pygame的具体使用讲解
Nov 03 Python
python复制文件到指定目录的实例
Apr 27 Python
详解Python3的TFTP文件传输
Jun 26 Python
Python中的Numpy矩阵操作
Aug 12 Python
python学生管理系统学习笔记
Mar 19 Python
使用Python的OpenCV模块识别滑动验证码的缺口(推荐)
May 10 Python
Python如何实现大型数组运算(使用NumPy)
Jul 24 Python
Python urlopen()参数代码示例解析
Dec 10 Python
Python获取、格式化当前时间日期的方法
Feb 10 #Python
Python3操作MongoDB增册改查等方法详解
Feb 10 #Python
Python实现结构体代码实例
Feb 10 #Python
Python守护进程实现过程详解
Feb 10 #Python
Python3 字典dictionary入门基础附实例
Feb 10 #Python
python列表返回重复数据的下标
Feb 10 #Python
Python中断多重循环的几种方式详解
Feb 10 #Python
You might like
php adodb介绍
2009/03/19 PHP
PHP中round()函数对浮点数进行四舍五入的方法
2014/11/19 PHP
PHP 实现的将图片转换为TXT
2015/10/21 PHP
详解PHP使用Redis存储session时的一个Warning定位
2017/07/05 PHP
PDO::commit讲解
2019/01/27 PHP
js实现无限级树形导航列表效果代码
2015/09/23 Javascript
jQuery easyui的validatebox校验规则扩展及easyui校验框validatebox用法
2016/01/18 Javascript
浅谈键盘上回车按钮的js触发事件
2017/02/13 Javascript
深入nodejs中流(stream)的理解
2017/03/27 NodeJs
Javascript中八种遍历方法的执行速度深度对比
2017/04/25 Javascript
jQuery UI实现动画效果代码分享
2018/08/19 jQuery
[10:07]2014DOTA2国际邀请赛 实拍选手现场观战DK对阵Titan
2014/07/12 DOTA
Python函数学习笔记
2008/10/07 Python
Python自动化测试ConfigParser模块读写配置文件
2016/08/15 Python
利用Python如何批量修改数据库执行Sql文件
2018/07/29 Python
在Pandas中DataFrame数据合并,连接(concat,merge,join)的实例
2019/01/29 Python
浅谈pyqt5中信号与槽的认识
2019/02/17 Python
python代理工具mitmproxy使用指南
2019/07/04 Python
如何用PyPy让你的Python代码运行得更快
2020/12/02 Python
python中_del_还原数据的方法
2020/12/09 Python
requests在python中发送请求的实例讲解
2021/02/17 Python
HTML5 input placeholder 颜色修改示例
2014/05/30 HTML / CSS
HTML5 Canvas中绘制椭圆的4种方法
2015/04/24 HTML / CSS
canvas像素点操作之视频绿幕抠图
2018/09/11 HTML / CSS
NIHAOMARKET官方海外旗舰店:意大利你好华人超市
2018/01/27 全球购物
SOA面试题:如何在SOA中实现松耦合
2013/07/21 面试题
人事部主管岗位职责
2013/12/26 职场文书
学生励志演讲稿
2014/01/06 职场文书
旷课检讨书2000字
2014/01/14 职场文书
给老师的一封建议书
2014/03/13 职场文书
英语教育专业自荐信
2014/05/29 职场文书
公务员学习习总书记“三严三实”思想汇报
2014/09/19 职场文书
2015年党员自我剖析材料
2014/12/17 职场文书
高中数学课堂教学反思
2016/02/18 职场文书
python在package下继续嵌套一个package
2022/04/14 Python
三星 3nm 芯片将于第二季度开始量产
2022/04/29 数码科技