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实现dict版图遍历示例
Feb 19 Python
python爬虫的工作原理
Mar 05 Python
Python通过future处理并发问题
Oct 17 Python
python简单图片操作:打开\显示\保存图像方法介绍
Nov 23 Python
Request的中断和ErrorHandler实例解析
Feb 12 Python
python 按不同维度求和,最值,均值的实例
Jun 28 Python
对Python3 序列解包详解
Feb 16 Python
Python判断有效的数独算法示例
Feb 23 Python
django rest framework vue 实现用户登录详解
Jul 29 Python
Python3实现将一维数组按标准长度分隔为二维数组
Nov 29 Python
PyCharm下载和安装详细步骤
Dec 17 Python
Python-numpy实现灰度图像的分块和合并方式
Jan 09 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+Javascript实现在线拍照功能实例
2015/07/18 PHP
图片之间的切换
2006/06/26 Javascript
将HTML自动转为JS代码
2006/06/26 Javascript
JQuery动态创建DOM、表单元素的实现代码
2011/08/09 Javascript
分页栏的web标准实现
2011/11/01 Javascript
js淡入淡出的图片轮播效果代码分享
2015/08/24 Javascript
Js得到radiobuttonlist选中值的两种方法(推荐)
2016/08/25 Javascript
JavaScript SHA1加密算法实现详细代码
2016/10/06 Javascript
轻松实现jquery选项卡切换效果
2016/10/10 Javascript
Javascript 实现放大镜效果实例详解
2016/12/03 Javascript
详解vue父子模版嵌套案例
2017/03/04 Javascript
ReactJs实现树形结构的数据显示的组件的示例
2017/08/18 Javascript
JS实现从对象获取对象中单个键值的方法示例
2019/06/05 Javascript
JavaScript使用百度ECharts插件绘制饼图操作示例
2019/11/26 Javascript
vue-autoui自匹配webapi的UI控件的实现
2020/03/20 Javascript
单线程JavaScript实现异步过程详解
2020/05/19 Javascript
[01:02:07]Liquid vs Newbee 2019国际邀请赛小组赛 BO2 第一场 8.15
2019/08/16 DOTA
Python3.x和Python2.x的区别介绍
2013/02/12 Python
理解python多线程(python多线程简明教程)
2014/06/09 Python
python抓取网页时字符集转换问题处理方案分享
2014/06/19 Python
python提示No module named images的解决方法
2014/09/29 Python
利用Python如何实现数据驱动的接口自动化测试
2018/05/11 Python
python 将md5转为16字节的方法
2018/05/29 Python
python多进程读图提取特征存npy
2019/05/21 Python
基于python plotly交互式图表大全
2019/12/07 Python
pytorch 状态字典:state_dict使用详解
2020/01/17 Python
利用OpenCV中对图像数据进行64F和8U转换的方式
2020/06/03 Python
如何使用localstorage代替cookie实现跨域共享数据问题
2018/04/18 HTML / CSS
Looking4Parking美国:全球排名第一的机场停车比较品牌
2019/08/26 全球购物
Carolina Lemke Berlin澳大利亚官网:时尚太阳镜品牌
2019/09/17 全球购物
雅虎笔试题(字符串操作)
2015/03/24 面试题
医药学专业大学生职业生涯规划书论文
2014/01/21 职场文书
九年级化学教学反思
2014/01/28 职场文书
全陪导游词
2015/02/04 职场文书
公积金贷款承诺书
2015/04/30 职场文书
html5 录制mp3音频支持采样率和比特率设置
2021/07/15 Javascript