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 中的变量名与变量值方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
下面小编就为大家分享一篇pandas数据框,统计某列数据对应的个数方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
这篇文章主要为大家介绍了python神经网络MobileNetV2模型的复现详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-05-05
这篇文章主要为大家详细介绍了python实现图像边缘检测，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-08-08
TensorFlow中tf.batch_matmul()的用法
这篇文章主要介绍了TensorFlow中tf.batch_matmul()的用法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-06-06
今天小编就为大家分享一篇python 使用plt画图,去除图片四周的白边方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
这篇文章主要给大家介绍了关于python自动生成证件号的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
在本篇内容里小编给大家整理的是关于python中取绝对值简单方法，需要的朋友们可以学习下。2020-07-07
这篇文章主要为大家详细介绍了python tornado微信开发入门代码，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-08-08
这篇文章主要为大家详细介绍了python编程实现维吉尼亚算法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-03-03
这篇文章主要介绍了Django 限制访问频率的思路详解，本文给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2019-12-12
最新评论