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实现全局变量的两个解决方法
Jul 03 Python
python多线程threading.Lock锁用法实例
Nov 01 Python
Python基础中所出现的异常报错总结
Nov 19 Python
Python判断两个对象相等的原理
Dec 12 Python
基于python 二维数组及画图的实例详解
Apr 03 Python
pandas 两列时间相减换算为秒的方法
Apr 20 Python
在Python中分别打印列表中的每一个元素方法
Nov 07 Python
Python变量类型知识点总结
Feb 18 Python
pyqt5实现绘制ui,列表窗口,滚动窗口显示图片的方法
Jun 20 Python
pytorch中 gpu与gpu、gpu与cpu 在load时相互转化操作
May 25 Python
在vscode中启动conda虚拟环境的思路详解
Dec 25 Python
Python 居然可以在 Excel 中画画你知道吗
Feb 15 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
thinkphp验证码显示不出来的解决方法
2014/03/29 PHP
PHP加密解密字符串汇总
2015/04/26 PHP
PHP实现对数组分页处理实例详解
2017/02/07 PHP
PHP cURL获取微信公众号access_token的实例
2018/04/28 PHP
PHP+ajax实现二级联动菜单功能示例
2018/08/10 PHP
PHP消息队列实现及应用详解【队列处理订单系统和配送系统】
2019/05/20 PHP
php设计模式之职责链模式实例分析【星际争霸游戏案例】
2020/03/27 PHP
javascript读取xml实现javascript分页
2013/12/13 Javascript
JS记录用户登录次数实现代码
2014/01/15 Javascript
调用jQuery滑出效果时闪烁的解决方法
2014/03/27 Javascript
JavaScript闭包函数访问外部变量的方法
2014/08/27 Javascript
Javascript中的数据类型之旅
2015/10/18 Javascript
JavaScript开发Chrome浏览器扩展程序UI的教程
2016/05/16 Javascript
AngularJS基础 ng-init 指令简单示例
2016/08/02 Javascript
AngularJS变量及过滤器Filter用法分析
2016/11/22 Javascript
基于Vue实现后台系统权限控制的示例代码
2017/08/29 Javascript
vue-cli3 从搭建到优化的详细步骤
2019/01/20 Javascript
Vue.js 中制作自定义选择组件的代码附演示demo
2020/02/28 Javascript
[00:43]DOTA2小紫本全民票选福利PA至宝全方位展示
2014/11/25 DOTA
Windows和Linux下使用Python访问SqlServer的方法介绍
2015/03/10 Python
python实现发送和获取手机短信验证码
2016/01/15 Python
Python实现将16进制字符串转化为ascii字符的方法分析
2017/07/21 Python
Python3 导入上级目录中的模块实例
2019/02/16 Python
python实现连续变量最优分箱详解--CART算法
2019/11/22 Python
windows下python 3.9 Numpy scipy和matlabplot的安装教程详解
2020/11/28 Python
详解Django关于StreamingHttpResponse与FileResponse文件下载的最优方法
2021/01/07 Python
CSS3中的Transition过度与Animation动画属性使用要点
2016/05/20 HTML / CSS
欧姆龙医疗欧洲有限公司:Omron Healthcare Europe B.V
2020/06/13 全球购物
农民入党思想汇报
2014/01/03 职场文书
公务员党员评议表自我鉴定
2014/09/14 职场文书
机关作风建设自查报告及整改措施
2014/10/21 职场文书
2015年城管个人工作总结范文
2015/04/20 职场文书
银行求职信怎么写
2019/06/20 职场文书
JS数组的常用方法整理
2021/03/31 Javascript
Vue + iView实现Excel上传功能的完整代码
2021/06/22 Vue.js
Python 中的Sympy详细使用
2021/08/07 Python