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实现全角半角转换的方法
Aug 18 Python
Python实现队列的方法
May 26 Python
python制作最美应用的爬虫
Oct 28 Python
深入理解python中函数传递参数是值传递还是引用传递
Nov 07 Python
让代码变得更易维护的7个Python库
Oct 09 Python
python爬虫获取小区经纬度以及结构化地址
Dec 30 Python
python 判断矩阵中每行非零个数的方法
Jan 26 Python
Python实现FTP弱口令扫描器的方法示例
Jan 31 Python
Python模拟FTP文件服务器的操作方法
Feb 18 Python
Python文字截图识别OCR工具实例解析
Mar 05 Python
Python任务自动化工具tox使用教程
Mar 17 Python
Python字符串三种格式化输出
Sep 17 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实现框架(二)
2006/10/09 PHP
ajax 的post方法实例(带循环)
2011/07/04 PHP
PHP里的中文变量说明
2011/07/23 PHP
PHP 500报错的快速解决方法
2016/12/14 PHP
jQuery 获取对象 定位子对象
2010/05/31 Javascript
离开页面时检测表单元素是否被修改,提示保存的js代码
2010/08/25 Javascript
jQuery在vs2008及js文件中的无智能提示的解决方法
2010/12/30 Javascript
js如何取消事件冒泡
2013/09/23 Javascript
jQuery实现的五子棋游戏实例
2015/06/13 Javascript
三种Node.js写文件的方式
2016/03/08 Javascript
JS基于面向对象实现的选项卡效果示例
2016/12/20 Javascript
JS实现定时任务每隔N秒请求后台setInterval定时和ajax请求问题
2017/10/15 Javascript
jquery点击回车键实现登录效果并默认焦点的方法
2018/03/09 jQuery
在小程序中使用腾讯视频插件播放教程视频的方法
2018/07/10 Javascript
Vue表单控件绑定图文详解
2019/02/11 Javascript
js中!和!!的区别与用法
2020/05/09 Javascript
JS轮播图的实现方法2
2020/08/25 Javascript
Appium+Python自动化测试之运行App程序示例
2019/01/23 Python
Python读写文件基础知识点
2019/06/10 Python
ORM Django 终端打印 SQL 语句实现解析
2019/08/09 Python
使用OpenCV实现仿射变换—缩放功能
2019/08/29 Python
python如果快速判断数字奇数偶数
2019/11/13 Python
Python结合百度语音识别实现实时翻译软件的实现
2021/01/18 Python
html5 Canvas画图教程(3)—canvas出现1像素线条模糊不清的原因
2013/01/09 HTML / CSS
Canvas 文本填充线性渐变的使用详解
2020/06/22 HTML / CSS
医学生实习自荐信
2013/10/01 职场文书
公积金单位接收函
2014/01/11 职场文书
大学生实习思想汇报
2014/01/12 职场文书
迎新晚会主持词开场白
2015/05/28 职场文书
办公室规章制度范本
2015/08/04 职场文书
创新创业项目计划书该怎样写?
2019/08/13 职场文书
导游词之包公祠
2019/11/25 职场文书
MySQL中使用or、in与union all在查询命令下的效率对比
2021/05/26 MySQL
mysql5.7使用binlog 恢复数据的方法
2021/06/03 MySQL
php解析非标准json、非规范json的方式实例
2022/05/10 PHP
mybatis 获取更新记录的id
2022/05/20 Java/Android