查看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