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中规范定义命名空间的一些建议
Jun 04 Python
python flask 多对多表查询功能
Jun 25 Python
使用Python的package机制如何简化utils包设计详解
Dec 11 Python
Python通用函数实现数组计算的方法
Jun 13 Python
Python判断字符串是否xx开始或结尾的示例
Aug 08 Python
python取均匀不重复的随机数方式
Nov 27 Python
python如何通过pyqt5实现进度条
Jan 20 Python
Pycharm连接远程服务器过程图解
Apr 30 Python
python的Jenkins接口调用方式
May 12 Python
基于python制作简易版学生信息管理系统
Apr 20 Python
python神经网络编程之手写数字识别
May 08 Python
Django实现drf搜索过滤和排序过滤
Jun 21 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 连接mysql连接被重置的解决方法
2011/02/15 PHP
使用php get_headers 判断URL是否有效的解决办法
2013/04/27 PHP
Avengerls vs Newbee BO3 第二场2.18
2021/03/10 DOTA
javascript实现的基于金山词霸网络翻译的代码
2010/01/15 Javascript
js word表格动态添加代码
2010/06/07 Javascript
JavaScript定义类或函数的几种方式小结
2011/01/09 Javascript
表单元素的submit()方法和onsubmit事件应用概述
2013/02/01 Javascript
javascript中的作用域和上下文使用简要概述
2013/12/05 Javascript
JavaScript获取页面上被选中文字的方法技巧
2015/03/13 Javascript
基于BootStrap的图片轮播效果展示实例代码
2016/05/23 Javascript
jQuery实现输入框邮箱内容自动补全与上下翻动显示效果【附demo源码下载】
2016/09/20 Javascript
关于Jquery中的事件绑定总结
2016/10/26 Javascript
令按钮悬浮在(手机)页面底部的实现方法
2017/05/02 Javascript
React-router4路由监听的实现
2018/08/07 Javascript
JavaScript实现的联动菜单特效示例
2019/07/08 Javascript
[52:37]完美世界DOTA2联赛循环赛 Forest vs DM BO2第一场 10.29
2020/10/29 DOTA
[01:36:17]DOTA2-DPC中国联赛 正赛 Ehome vs iG BO3 第一场 1月31日
2021/03/11 DOTA
Python类属性的延迟计算
2016/10/22 Python
Python实现自动发送邮件功能
2021/03/02 Python
书单|人生苦短,你还不用python!
2017/12/29 Python
python pandas读取csv后,获取列标签的方法
2018/11/12 Python
利用Python校准本地时间的方法教程
2019/10/31 Python
Python简易计算器制作方法代码详解
2019/10/31 Python
Spring实战之使用util:命名空间简化配置操作示例
2019/12/09 Python
Python 模拟动态产生字母验证码图片功能
2019/12/24 Python
python xlsxwriter模块的使用
2020/12/24 Python
意大利香水和彩妆护肤品购物网站:Ditano
2017/08/13 全球购物
科室工作个人总结的自我评价
2013/10/29 职场文书
老干部工作先进事迹
2014/08/17 职场文书
数学教师个人总结
2015/02/06 职场文书
工伤劳动仲裁代理词
2015/05/25 职场文书
飞屋环游记观后感
2015/06/08 职场文书
首次购房证明
2015/06/19 职场文书
2016简历自荐信优秀范文
2016/01/29 职场文书
《自己的花是让别人看的》教学反思
2016/02/19 职场文书
Python采集股票数据并制作可视化柱状图
2022/04/04 Python