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实现批量把SVG格式转成png、pdf格式的代码分享
Aug 21 Python
浅谈python装饰器探究与参数的领取
Dec 01 Python
python调用xlsxwriter创建xlsx的方法
May 03 Python
python使用turtle库绘制树
Jun 25 Python
python utc datetime转换为时间戳的方法
Jan 15 Python
详解Python中is和==的区别
Mar 21 Python
pyqt5 删除layout中的所有widget方法
Jun 25 Python
Python实现检测文件的MD5值来查找重复文件案例
Mar 12 Python
python 爬取免费简历模板网站的示例
Sep 27 Python
Python数据可视化常用4大绘图库原理详解
Oct 23 Python
termux中matplotlib无法显示中文问题的解决方法
Jan 11 Python
python状态机transitions库详解
Jun 02 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设计模式之责任链模式的深入解析
2013/06/13 PHP
ThinkPHP 3.2.2实现事务操作的方法
2017/05/05 PHP
PHP实践教程之过滤、验证、转义与密码详解
2017/07/24 PHP
PHP cookie,session的使用与用户自动登录功能实现方法分析
2019/06/05 PHP
PHP基于array_unique实现二维数组去重
2020/07/14 PHP
使用Modello编写JavaScript类
2006/12/22 Javascript
基于jQuery的图片大小自动适应实现代码
2010/11/17 Javascript
javascript实现的一个带下拉框功能的文本框
2014/05/08 Javascript
深入学习jQuery Validate表单验证(二)
2016/01/18 Javascript
Jquery判断form表单数据是否变化
2016/03/30 Javascript
javascript学习之json入门
2016/12/22 Javascript
实例解析js中try、catch、finally的执行规则
2017/02/24 Javascript
详解Node.js中exports和module.exports的区别
2017/04/19 Javascript
详解nodejs微信jssdk后端接口
2017/05/25 NodeJs
webpack 单独打包指定JS文件的方法
2018/02/22 Javascript
微信小程序实现跳转的几种方式总结(推荐)
2019/04/24 Javascript
微信网页登录逻辑与实现方法
2019/04/29 Javascript
[02:32]DOTA2英雄基础教程 祸乱之源
2013/12/23 DOTA
Python是编译运行的验证方法
2015/01/30 Python
利用numpy和pandas处理csv文件中的时间方法
2018/04/19 Python
Python minidom模块用法示例【DOM写入和解析XML】
2019/03/25 Python
使用python PIL库实现简单验证码的去噪方法步骤
2019/05/10 Python
使用python接入微信聊天机器人
2020/03/31 Python
python实现文件的分割与合并
2019/08/29 Python
CSS3 特效范例整理
2011/08/22 HTML / CSS
英国在线玫瑰专家:InterRose
2019/12/01 全球购物
介绍一下Ruby的多线程处理
2013/02/01 面试题
音乐系毕业生自荐信
2013/10/27 职场文书
中学教师培训制度
2014/01/31 职场文书
工商管理本科生求职信
2014/07/13 职场文书
夫妻分居协议书范文
2014/11/26 职场文书
给老婆的检讨书
2015/01/27 职场文书
办公室主任岗位职责范本
2015/03/31 职场文书
2016年世界艾滋病日宣传活动总结
2016/04/01 职场文书
vscode中使用npm安装babel的方法
2021/08/02 Javascript
python垃圾回收机制原理分析
2022/04/13 Python