查看TensorFlow checkpoint文件中的变量名和对应值方法
更新时间：2018年06月14日 10:15:55 投稿：jingxian
今天小编就为大家分享一篇查看TensorFlow checkpoint文件中的变量名和对应值方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
实例如下所示：
from tensorflow.python import pywrap_tensorflow
checkpoint_path = os.path.join(model_dir, "model.ckpt")
reader = pywrap_tensorflow.NewCheckpointReader(checkpoint_path)
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)) # Remove this is you want to print only variable names
以上这篇查看TensorFlow checkpoint文件中的变量名和对应值方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
使用 Python 可以使用机器学习模型进行温度预测。常用的模型有回归分析、随机森林等。本文就来和大家来了具体实现方法，希望对大家有所帮助2023-01-01
今天小编就为大家分享一篇TensorFlow设置日志级别的几种方式小结，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02
这篇文章主要介绍了Python 入门学习之函数式编程， Python 中的函数式编程技术进行了简单的入门介绍，下文详细内容需要的小伙伴可以参考一下2022-05-05
基于python实现自动化办公学习笔记(CSV、word、Excel、PPT)
这篇文章主要介绍了基于python实现自动化办公学习笔记，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-08-08
在本文中我们系统的给大家整理了关于python字符串和常用数据结构的相关知识点以及实例代码，需要的朋友们学习下。2019-05-05
今天小编就为大家分享一篇python提取具有某种特定字符串的行数据方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
今天小编就为大家分享一篇python画双y轴图像的示例代码，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
modelform是django中基于已有的model,快速生成的具有form功能的一个模块,下面这篇文章主要给大家介绍了关于Django中ModelForm组件的简单配置与使用教程,需要的朋友可以参考下2022-10-10
这篇文章主要为大家详细介绍了Python中图像通用操作的实现，例如：图像旋转、图像缩放等，文中的示例代码讲解详细，需要的可以参考一下2023-07-07
这篇文章主要介绍了使用wxpython实现的一个简单图片浏览器实例,根据自己的特殊需求而写,需要的朋友可以参考下2014-07-07
最新评论