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编写函数装饰器
Mar 18 Python
python编程线性回归代码示例
Dec 07 Python
通过Py2exe将自己的python程序打包成.exe/.app的方法
May 26 Python
Python2.7.10以上pip更新及其他包的安装教程
Jun 12 Python
Python 查找list中的某个元素的所有的下标方法
Jun 27 Python
python 重命名轴索引的方法
Nov 10 Python
Python多进程写入同一文件的方法
Jan 14 Python
Python 编程速成(推荐)
Apr 15 Python
python属于软件吗
Jun 18 Python
Python descriptor(描述符)的实现
Nov 15 Python
用Python实现一个打字速度测试工具来测试你的手速
May 28 Python
Python读取和写入Excel数据
Apr 20 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字符串截取中文截取2,单字节截取模式
2007/12/10 PHP
mysql From_unixtime及UNIX_TIMESTAMP及DATE_FORMAT日期函数
2010/03/21 PHP
PHP利用header跳转失效的解决方法
2014/10/24 PHP
AES加解密在php接口请求过程中的应用示例
2016/10/26 PHP
JSON 教程 json入门学习笔记
2020/09/22 Javascript
Javascript实现Web颜色值转换
2015/02/05 Javascript
详解jQuery简单的表格应用
2016/12/16 Javascript
JS调用Android、Ios原生控件
2017/01/06 Javascript
微信小程序 详解页面跳转与返回并回传数据
2017/02/13 Javascript
js实现动态显示时间效果
2017/03/06 Javascript
AugularJS从入门到实践(必看篇)
2017/07/10 Javascript
Vue 2.0双向绑定原理的实现方法
2019/10/23 Javascript
简单了解常用的JavaScript 库
2020/07/16 Javascript
[13:56]DAC2018 4.5SOLO赛决赛 MidOne vs Paparazi第一场
2018/04/06 DOTA
在Windows8上的搭建Python和Django环境
2014/07/03 Python
使用Python脚本对Linux服务器进行监控的教程
2015/04/02 Python
Python通过90行代码搭建一个音乐搜索工具
2015/07/29 Python
使用Python获取并处理IP的类型及格式方法
2018/11/01 Python
PyQt5基本控件使用详解:单选按钮、复选框、下拉框
2019/08/05 Python
pytorch中tensor张量数据类型的转化方式
2019/12/31 Python
python通用读取vcf文件的类(复制粘贴即可用)
2020/02/29 Python
在脚本中单独使用django的ORM模型详解
2020/04/01 Python
Jupyter Notebook远程登录及密码设置操作
2020/04/10 Python
Python命令行参数定义及需要注意的地方
2020/11/30 Python
励志演讲稿800字
2014/08/21 职场文书
教师工作总结范文2014
2014/11/10 职场文书
先进党组织事迹材料
2014/12/26 职场文书
自愿离婚协议书范本
2015/01/26 职场文书
幼儿园中班个人总结
2015/02/28 职场文书
2015年乡镇纪检工作总结
2015/04/22 职场文书
2016年暑期见闻作文
2015/11/25 职场文书
教你怎么用python实现字符串转日期
2021/05/24 Python
解决Laravel使用验证时跳转到首页的问题
2021/11/17 PHP
Java中try catch处理异常示例
2021/12/06 Java/Android
JavaScript实现外溢动态爱心的效果的示例代码
2022/03/21 Javascript
netty 实现tomcat的示例代码
2022/06/05 Servers