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中使用enumerate函数遍历元素实例
Jun 16 Python
Python中的is和id用法分析
Jan 26 Python
python3简单实现微信爬虫
Apr 09 Python
简单的Python的curses库使用教程
Apr 11 Python
深入理解NumPy简明教程---数组2
Dec 17 Python
基于Python os模块常用命令介绍
Nov 03 Python
详解Python list和numpy array的存储和读取方法
Nov 06 Python
VSCode中自动为Python文件添加头部注释
Nov 14 Python
Python使用matplotlib绘制圆形代码实例
May 27 Python
使用Python绘制台风轨迹图的示例代码
Sep 21 Python
Python+unittest+requests+excel实现接口自动化测试框架
Dec 23 Python
4种方法python批量修改替换列表中元素
Apr 07 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流量统计功能的实现代码
2012/09/29 PHP
编译php 5.2.14+fpm+memcached(具体操作详解)
2013/06/18 PHP
JavaScript实现班级随机点名小应用需求的具体分析
2014/05/12 Javascript
无限树Jquery插件zTree的常用功能特性总结
2014/09/11 Javascript
js表格排序实例分析(支持int,float,date,string四种数据类型)
2015/05/06 Javascript
jquery——九宫格大转盘抽奖实例
2017/01/16 Javascript
你真的了解BOM中的history对象吗
2017/02/13 Javascript
JavaScript中的this陷阱的最全收集并整理(没有之一)
2017/02/21 Javascript
javascript浏览器用户代理检测脚本实现方法
2017/10/27 Javascript
通过vue-cli来学习修改Webpack多环境配置和发布问题
2017/12/22 Javascript
JavaScript实现多张图片放大镜效果示例【不限定图片尺寸,rem单位】
2019/05/14 Javascript
vue组件 keep-alive 和 transition 使用详解
2019/10/11 Javascript
微信小程序跨页面数据传递事件响应实现过程解析
2019/12/19 Javascript
js实现蒙版效果
2020/01/11 Javascript
vue实现评价星星功能
2020/06/30 Javascript
jquery实现广告上下滚动效果
2021/03/04 jQuery
[39:46]完美世界DOTA2联赛PWL S2 LBZS vs Rebirth 第二场 11.25
2020/11/25 DOTA
[32:07]完美世界DOTA2联赛PWL S3 LBZS vs Rebirth 第一场 12.16
2020/12/17 DOTA
Python编程语言的35个与众不同之处(语言特征和使用技巧)
2014/07/07 Python
python编程实现归并排序
2017/04/14 Python
浅谈机器学习需要的了解的十大算法
2017/12/15 Python
python smtplib模块自动收发邮件功能(一)
2018/05/22 Python
解决pycharm界面不能显示中文的问题
2018/05/23 Python
详解Django CAS 解决方案
2019/10/30 Python
python不到50行代码完成了多张excel合并的实现示例
2020/05/28 Python
Python如何输出整数
2020/06/07 Python
英国领先的瓷砖专家:Walls and Floors
2018/04/27 全球购物
财会自我鉴定范文
2013/12/27 职场文书
擅自离岗检讨书
2014/02/11 职场文书
商铺门前三包责任书
2014/07/25 职场文书
2015年七年级班主任工作总结
2015/05/21 职场文书
2015年节能降耗工作总结
2015/05/22 职场文书
大学生暑假实习总结
2015/07/13 职场文书
60条职场经典语录,总有一条能触动你的心
2019/08/21 职场文书
python基础入门之普通操作与函数(三)
2021/06/13 Python
python疲劳驾驶困倦低头检测功能的实现
2022/04/04 Python