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使用wmi模块获取windows下硬盘信息的方法
May 15 Python
Python while、for、生成器、列表推导等语句的执行效率测试
Jun 03 Python
使用Python编写基于DHT协议的BT资源爬虫
Mar 19 Python
python中的错误处理
Apr 10 Python
python实现字典(dict)和字符串(string)的相互转换方法
Mar 01 Python
详解python里使用正则表达式的分组命名方式
Oct 24 Python
用python 批量更改图像尺寸到统一大小的方法
Mar 31 Python
python读取tif图片时保留其16bit的编码格式实例
Jan 13 Python
python基于property()函数定义属性
Jan 22 Python
利用keras加载训练好的.H5文件,并实现预测图片
Jan 24 Python
Python dict和defaultdict使用实例解析
Mar 12 Python
68行Python代码实现带难度升级的贪吃蛇
Jan 18 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源代码
2009/08/21 PHP
ThinkPHP之foreach标签使用概述
2014/06/30 PHP
php通过array_merge()函数合并两个数组的方法
2015/03/18 PHP
tp5框架前台无限极导航菜单类实现方法分析
2020/03/29 PHP
dojo 之基础篇
2007/03/24 Javascript
Tab切换组件(选项卡功能)实例代码
2013/11/21 Javascript
2014最热门的JavaScript代码高亮插件推荐
2014/11/25 Javascript
jquery.cookie.js使用指南
2015/01/05 Javascript
jQuery中DOM树操作之复制元素的方法
2015/01/23 Javascript
JavaScript数组去重的3种方法和代码实例
2015/07/01 Javascript
JavaScript常用字符串与数组扩展函数小结
2016/04/24 Javascript
JS使用单链表统计英语单词出现次数
2016/06/16 Javascript
jQuery实现点击某个div打开层,点击其他div关闭层实例分析(阻止冒泡)
2016/11/18 Javascript
微信小程序 配置文件详细介绍
2016/12/14 Javascript
bootstrap table之通用方法( 时间控件,导出,动态下拉框, 表单验证 ,选中与获取信息)代码分享
2017/01/24 Javascript
详解bootstrap的modal-remote两种加载方式【强化】
2017/01/27 Javascript
jquery实现下拉框左右选择功能
2017/02/21 Javascript
VueJS如何引入css或者less文件的一些坑
2017/04/25 Javascript
react-native使用react-navigation进行页面跳转导航的示例
2017/09/07 Javascript
javascript 面向对象实战思想分享
2017/09/07 Javascript
Angularjs渲染的 using 指令的星级评分系统示例
2017/11/09 Javascript
[52:37]完美世界DOTA2联赛循环赛 Forest vs DM BO2第一场 10.29
2020/10/29 DOTA
Python中的高级数据结构详解
2015/03/27 Python
对numpy中布尔型数组的处理方法详解
2018/04/17 Python
解决Python print输出不换行没空格的问题
2018/11/14 Python
python-opencv获取二值图像轮廓及中心点坐标的代码
2019/08/27 Python
给你一面国旗 教你用python画中国国旗
2019/09/24 Python
德国户外装备、登山运动和攀岩商店:tapir store
2020/02/12 全球购物
Happy Plugs官网:瑞典无线耳机品牌
2020/07/16 全球购物
C语言变量的命名规则都有哪些
2013/12/27 面试题
什么是ESB?请介绍一下ESB?
2015/05/27 面试题
好习惯伴我成长演讲稿
2014/05/21 职场文书
王金山在党的群众路线教育实践活动总结大会上的讲话稿
2014/10/25 职场文书
机器人总动员观后感
2015/06/09 职场文书
python3 删除所有自定义变量的操作
2021/04/08 Python
SpringBoot中HttpSessionListener的简单使用方式
2022/03/17 Java/Android