查看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数据可视化之简单折线图的绘制

    Python数据可视化之简单折线图的绘制

    这篇文章主要为大家详细介绍了Python数据可视化之绘制简单折线图的相关资料,文中的示例代码简洁易懂,感兴趣的小伙伴可以了解一下
    2022-10-10
  • python 动态规划问题解析(背包问题和最长公共子串)

    python 动态规划问题解析(背包问题和最长公共子串)

    这篇文章主要介绍了python 动态规划(背包问题和最长公共子串),在动态规划中,你要将某个指标最大化。在这个例子中,你要找出两个单词的最长公共子串。fish和fosh都包含的最长子串是什么呢,感兴趣的朋友跟随小编一起看看吧
    2022-05-05
  • pytorch 批次遍历数据集打印数据的例子

    pytorch 批次遍历数据集打印数据的例子

    今天小编就为大家分享一篇pytorch 批次遍历数据集打印数据的例子,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2019-12-12
  • python 包之 re 正则匹配教程分享

    python 包之 re 正则匹配教程分享

    这篇文章主要介绍了python 包之 re 正则匹配教程分享,文章基于python 包 re的相关资料展开贵主题的详细介绍,需要的小伙伴可以参考一下
    2022-04-04
  • Python入门之基础语法详解

    Python入门之基础语法详解

    这篇文章主要介绍了Python入门之基础语法详解,文中有非常详细的代码示例,对正在学习python的小伙伴们有很大的帮助哟,需要的朋友可以参考下
    2021-05-05
  • Python基本语法之运算符功能与用法详解

    Python基本语法之运算符功能与用法详解

    这篇文章主要介绍了Python基本语法之运算符功能与用法,结合实例形式总结分析了Python常见运算符相关使用技巧与操作注意事项,需要的朋友可以参考下
    2019-10-10
  • Python批量修改文件名,文件再多也只要一秒

    Python批量修改文件名,文件再多也只要一秒

    大家好,本篇文章主要讲的是Python批量修改文件名,文件再多也只要一秒,感兴趣的同学赶快来看一看,对你有帮助的话记得收藏一下,方便下次浏览
    2021-12-12
  • python使用difflib实现自动查重

    python使用difflib实现自动查重

    Python中有许多现成的库和工具,可以方便地实现自动查重的功能,其中,difflib库就是一个专门用于比较文件和字符串差异的库,下面我们就来看看如何利用difflib实现自动查重吧
    2023-11-11
  • 一次python-flask蓝图的踩坑记录

    一次python-flask蓝图的踩坑记录

    简单来说,蓝图就是一个存储操作路由映射方法的容器,主要用来实现客户端请求和URL相互关联的功能,在Flask中,使用蓝图可以帮助我们实现模块化应用的功能,这篇文章主要给大家介绍了关于python-flask蓝图踩坑的相关资料,需要的朋友可以参考下
    2021-08-08
  • python pygame实现方向键控制小球

    python pygame实现方向键控制小球

    这篇文章主要为大家详细介绍了python pygame实现方向键控制小球,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-05-05

最新评论