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中的RSS处理
Apr 13 Python
Python Sql数据库增删改查操作简单封装
Apr 18 Python
python+mongodb数据抓取详细介绍
Oct 25 Python
python正则表达式re之compile函数解析
Oct 25 Python
python实现机械分词之逆向最大匹配算法代码示例
Dec 13 Python
python的numpy模块安装不成功简单解决方法总结
Dec 23 Python
python 弹窗提示警告框MessageBox的实例
Jun 18 Python
在Python中合并字典模块ChainMap的隐藏坑【推荐】
Jun 27 Python
Python基本类型的连接组合和互相转换方式(13种)
Dec 16 Python
pytorch forward两个参数实例
Jan 17 Python
PyTorch的torch.cat用法
Jun 28 Python
Python numpy矩阵处理运算工具用法汇总
Jul 13 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 xfocus防注入资料
2008/04/27 PHP
PHP array操作10个小技巧分享
2011/06/23 PHP
PHP return语句的另一个作用
2014/07/30 PHP
使用PHP和JavaScript判断请求是否来自微信内浏览器
2015/08/18 PHP
Yii框架实现多数据库配置和操作的方法
2017/05/25 PHP
PHP守护进程化在C和PHP环境下的实现
2017/11/21 PHP
setTimeout与setInterval在不同浏览器下的差异
2010/01/24 Javascript
一个很简单的jquery+xml+ajax的无刷新树结构(无css,后台是c#)
2010/06/02 Javascript
IE中jquery.form中ajax提交没反应解决方法分享
2012/09/11 Javascript
JavaScript基础教程之alert弹出提示框实例
2014/10/16 Javascript
js clearInterval()方法的定义和用法
2015/11/11 Javascript
跟我学习javascript的全局变量
2015/11/16 Javascript
深入理解JavaScript函数参数(推荐)
2016/07/26 Javascript
JavaScript实现星星等级评价功能
2017/03/22 Javascript
js 用于检测类数组对象的函数方法
2017/05/02 Javascript
深究AngularJS——ng-checked(回写:带真实案例代码)
2017/06/13 Javascript
Angular中的$watch、$watchGroup、$watchCollection
2017/06/25 Javascript
three.js实现3D影院的原理的代码分析
2017/12/18 Javascript
JavaScript原型链与继承操作实例总结
2018/08/24 Javascript
移动端滑动切换组件封装 vue-swiper-router实例详解
2018/11/25 Javascript
JavaScript继承与聚合实例详解
2019/01/22 Javascript
Vue使用Canvas绘制图片、矩形、线条、文字,下载图片
2019/04/26 Javascript
JavaScript图片旋转效果实现方法详解
2020/06/28 Javascript
python解析html开发库pyquery使用方法
2014/02/07 Python
深入理解Python单元测试unittest的使用示例
2017/11/18 Python
python通过百度地图API获取某地址的经纬度详解
2018/01/28 Python
利用Pyhton中的requests包进行网页访问测试的方法
2018/12/26 Python
OpenCV+Python识别车牌和字符分割的实现
2019/01/31 Python
mac使用python识别图形验证码功能
2020/01/10 Python
pytorch实现mnist分类的示例讲解
2020/01/10 Python
pandas处理csv文件的方法步骤
2020/10/16 Python
canvas裁剪clip()函数的具体使用
2018/03/01 HTML / CSS
沙龙级头发造型工具:FOXYBAE
2018/07/01 全球购物
美国婴儿和儿童服装购物网站:PatPat
2020/10/01 全球购物
构造器Constructor是否可被override?
2013/08/06 面试题
四十年同学聚会致辞
2015/07/28 职场文书