tensorflow实现打印ckpt模型保存下的变量名称及变量值
作者：Shwan_Ma 发布时间：2021-07-02 21:19:25
标签：tensorflow,ckpt,变量名称,变量值
有时候会需要通过从保存下来的ckpt文件来观察其保存下来的训练完成的变量值。
ckpt文件名列表：（一般是三个文件）
xxxxx.ckpt.data-00000-of-00001
xxxxx.ckpt.index
xxxxx.ckpt.meta
import os
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join("文件夹路径", "xxxxx.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path) #tf.train.NewCheckpointReader
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print("tensor_name: ", key)
# print(reader.get_tensor(key))
来源：https://blog.csdn.net/shwan_ma/article/details/84941051
0
投稿猜你喜欢
- 目录一、进程的创建1、一些常用方法介绍二、进程池的使用三、多进程和多线程的优缺点对比一、进程的创建Python的multiprocessin
- sys 模块提供了许多函数和变量来处理 Python 运行时环境的不同部分.处理命令行参数在解释器启动后, argv 列表包含了传递给脚本的
- username1: "(^[a-zA-Z]{1}([a-zA-Z0-9_]){4,14}|(^[\u4E00-\uFA29]{1
一、优先级队列介绍我们知道，普通的队列插入一个元素，数据会被放在后端，并且需要前面所有的元素都处理完成后才会处理前面的数据。但是优先级队列，- 本文实例讲述了Python socket实现的简单通信功能。分享给大家供大家参考，具体如下：套接字(socket)是计算机网络数据结构，在任
- 我就废话不多说了，大家还是直接看代码吧！#先定义两个矩阵X=np.array([[1,2104,5,1,45],[1,1416,3,2,40
- 做设计的时候，如何配色是经常让人头痛的问题，尽管很多时候，很多人都说对于色彩的掌握更多的是靠感觉，但是不得不说，感觉也是要有依据的。所以颜色
- 在写登录功能的时候看着网上的视频学着做,然后看了源码的时候发现了一些有意思的功能,因此写这一篇笔记分享给大家.1.阅读Django自带用户系
- 本文实例讲述了Python随机生成彩票号码的方法。分享给大家供大家参考。具体如下：前些日子在淘宝上买了一阵子彩票，每次都是使用淘宝的机选，每
- 漂亮的代码是漂亮网站的基础，优秀的 CSS 只存在与同样优秀的 HTML 之上，干净的，语义的 HTML 代码让一个网站更健壮。本文讲述了1
- 一、简单介绍pip 是 Python 包管理工具，该工具提供了对Python 包的查找、下载、安装和卸载的功能，现在大家用到的所有包不是自带
- 1. 介绍通俗的来讲，MobaXterm就是一款SSH客户端，它帮助我们在Windows操作系统下去连接并操作Linux服务器。MobaXt
- 很多现代的浏览器在地址栏的右边有个搜索框，默认的安装有 Google 搜索等。如下图所示：其实这是 OpenSearch 的一个应用，只要编
- 您是否知道 OpenCV 具有执行行人检测的内置方法？OpenCV 附带一个预训练的 HOG + 线性 SVM 模型，可用于在图像和视频流中
- 内容摘要合理使用渐变留白网格布局提高字体应用明确而有效的导航设计漂亮、有用的页脚介绍优秀设计和卓越设计之间的区别是比较小的。一般人可能无法解
- 本文实例讲述了python提取页面内url列表的方法。分享给大家供大家参考。具体实现方法如下：from bs4 import Beautif
- 今天看看那些知名的Logo，可能你没注意过，它们也是一直在变化，有的还挺雷人的！先从时尚的苹果开始吧，苹果也有土的时候。1976年那第一个L
- 如下所示：matplotlib subplots 设置总图的标题 ：fig.suptitle(dname,fontsize=16,x=0.5
- 在Pytorch中，torch.utils.data中的Dataset与DataLoader是处理数据集的两个函数，用来处理加载数据集。通常
- 这一块的内容很少, 异常使用try except即可, 日志只需要几行配置.使用装饰器捕获方法内的所有异常我使用装饰器来整个包裹一个方法,
