TensorFlow 输出checkpoint 中的变量名与变量值方式
更新时间：2020年02月11日 14:40:54 作者：qq_32799915
今天小编就为大家分享一篇TensorFlow 输出checkpoint 中的变量名与变量值方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
废话不多说，直接看代码吧！
import os
from tensorflow.python import pywrap_tensorflow
model_dir="/xxxxxxxxx/model.ckpt" #checkpoint的文件位置
# Read data from checkpoint file
reader = pywrap_tensorflow.NewCheckpointReader(model_dir)
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)) #输出变量值
输出结果：
这里只输出了变量名
以上这篇TensorFlow 输出checkpoint 中的变量名与变量值方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家详细介绍了python调用Matplotlib绘制分布点图，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-10-10
这篇文章主要介绍了使用Python的pencolor函数实现渐变色功能,本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-03-03
今天小编就为大家分享一篇python-序列解包(对可迭代元素的快速取值方法)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
这篇文章主要介绍了Python中函数带括号和不带括号的区别及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
在日常工作和学习中,我们经常需要比较两个文本文件之间的差异,本文将介绍如何使用Python编写一个文本比较工具,感兴趣的小伙伴可以了解一下2025-03-03
这篇文章主要为大家详细介绍了python实现决策树的相关代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2017-12-12
这篇文章主要介绍了关于Python调用百度语音合成SDK实现文字转音频的方法,AipSpeech是语音合成的Python SDK客户端，为使用语音合成的开发人员提供了一系列的交互方法,需要的朋友可以参考下2023-07-07
这篇文章主要介绍了python抓取网页内容示例,在抓取的时候对于gbk编码网页还需要转化一下，具体看下面的示例吧2014-02-02
今天小编就为大家分享一篇pytorch神经网络之卷积层与全连接层参数的设置方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
在Python的Django框架的视图中使用Session的方法
这篇文章主要介绍了在Python的Django框架的视图中使用Session的方法,包括相关的设置测试Cookies的方法,需要的朋友可以参考下2015-07-07
最新评论