查看TensorFlow checkpoint文件中的变量名和对应值方法

作者:jingxian 时间:2022-07-25 08:50:17 

实例如下所示:


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

来源:https://stackoverflow.com/questions/38218174/how-to-find-the-variable-names-that-are-saved-in-a-tensorflow-checkpoint/38226516#38226516

标签:TensorFlow,checkpoint
0
投稿

猜你喜欢

  • 关于Python中字符串的各种操作

    2022-06-25 16:18:38
  • MySQL中的count(*) 和 count(1) 区别性能对比分析

    2024-01-24 03:12:51
  • pandas 条件搜索返回列表的方法

    2022-09-16 14:46:34
  • python实现自动抢课脚本的示例代码

    2022-03-12 08:16:19
  • Oracle数据安全面面观

    2024-01-20 02:12:38
  • 关于python爬虫应用urllib库作用分析

    2023-11-02 12:59:43
  • 正确理解python迭代器与生成器

    2022-09-14 22:55:57
  • Python 创建子进程模块subprocess详解

    2022-02-21 06:47:39
  • MySQL delete删除数据后释放磁盘空间的操作方法

    2024-01-18 21:00:16
  • 使用get方式提交表单在地址栏里面不显示提交信息

    2024-06-05 09:35:10
  • 只用40行Python代码就能写出pdf转word小工具

    2021-11-03 18:02:00
  • 详解使用uni-app开发微信小程序之登录模块

    2024-05-11 09:15:57
  • 使用Python编写一个在Linux下实现截图分享的脚本的教程

    2022-04-23 13:23:11
  • 使用Vue-router二级路由跳转另一条路由下的子级

    2024-05-09 09:52:07
  • MySQL详解如何优化查询条件

    2024-01-26 06:52:05
  • 又一个php 分页类实现代码

    2023-11-15 13:04:56
  • mysql 10w级别的mysql数据插入

    2024-01-16 01:42:46
  • jquery和css3中的选择器nth-child使用方法和用途示例

    2024-04-25 13:11:35
  • mysql与mssql的md5加密语句

    2024-01-20 07:33:22
  • Perl集群配置管理系统Rex简明手册

    2023-07-15 20:00:34
  • asp之家 网络编程 m.aspxhome.com