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 BeautifulSoup设置页面编码的方法
Apr 03 Python
Python通过正则表达式选取callback的方法
Jul 18 Python
python中print的不换行即时输出的快速解决方法
Jul 20 Python
Python实现PS图像明亮度调整效果示例
Jan 23 Python
使用python爬取B站千万级数据
Jun 08 Python
根据DataFrame某一列的值来选择具体的某一行方法
Jul 03 Python
利用ctypes获取numpy数组的指针方法
Feb 12 Python
Python面向对象程序设计类变量与成员变量、类方法与成员方法用法分析
Apr 12 Python
Python Pandas实现数据分组求平均值并填充nan的示例
Jul 04 Python
解决numpy矩阵相减出现的负值自动转正值的问题
Jun 03 Python
Python 发送邮件方法总结
Aug 10 Python
总结Python使用过程中的bug
Jun 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 JSON 数据解析代码
2010/05/26 PHP
PHP 实现重载
2021/03/09 PHP
十分钟打造AutoComplete自动完成效果代码
2009/12/26 Javascript
js多级树形弹出一个小窗口层(非常好用)实例代码
2013/03/19 Javascript
JS利用cookie记忆当前位置的防刷新导航效果
2015/10/15 Javascript
微信小程序 教程之wxapp 视图容器 view
2016/10/19 Javascript
AngularJS学习笔记(三)数据双向绑定的简单实例
2016/11/08 Javascript
JavaScript正则表达式exec/g实现多次循环用法示例
2017/01/17 Javascript
AngularJS+bootstrap实现动态选择商品功能示例
2017/05/17 Javascript
AngularJS实现自定义指令与控制器数据交互的方法示例
2017/06/19 Javascript
js+html5实现页面可刷新的倒计时效果
2017/07/15 Javascript
vue中eventbus被多次触发以及踩过的坑
2017/12/02 Javascript
vue项目base64字符串转图片的实现代码
2018/07/13 Javascript
微信小程序开发之路由切换页面重定向问题
2018/09/18 Javascript
Layui实现带查询条件的分页
2019/07/27 Javascript
原生js实现3D轮播图
2020/03/21 Javascript
JQuery省市联动效果实现过程详解
2020/05/08 jQuery
Vue单文件组件开发实现过程详解
2020/07/30 Javascript
python组合无重复三位数的实例
2018/11/13 Python
在Python中字典根据多项规则排序的方法
2019/01/21 Python
Python 实现两个服务器之间文件的上传方法
2019/02/13 Python
Python 限定函数参数的类型及默认值方式
2019/12/24 Python
Python文件名匹配与文件复制的实现
2020/12/11 Python
一款纯css3实现的颜色渐变按钮的代码教程
2014/11/12 HTML / CSS
HTML5本地存储localStorage、sessionStorage基本用法、遍历操作、异常处理等
2014/05/08 HTML / CSS
关于canvas.toDataURL 在iOS运行失败的问题解决
2020/09/16 HTML / CSS
草莓网英国官网:Strawberrynet UK
2017/02/12 全球购物
碧欧泉法国官网:Biotherm法国
2019/10/23 全球购物
经济管理毕业生求职信
2014/03/15 职场文书
2014年前台文员工作总结
2014/12/08 职场文书
主持人开幕词
2015/01/29 职场文书
2015年纪念“卢沟桥事变”78周年活动方案
2015/05/06 职场文书
2016五四青年节活动总结范文
2016/04/06 职场文书
Html5大屏数据可视化开发的实现
2021/06/11 HTML / CSS
Java使用httpRequest+Jsoup爬取红蓝球号码
2021/07/02 Java/Android
Vue操作Storage本地化存储
2022/04/29 Vue.js