tensorflow查看ckpt各节点名称实例
作者:深夜虫鸣 时间:2021-03-26 14:04:49
运行下列脚本,可以打印出模型各个节点变量的名称:
from tensorflow.python import pywrap_tensorflow
import os
checkpoint_path=os.path.join('model.ckpt-131805')
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
checkpoint_path为自己的模型路径
来源:https://blog.csdn.net/u010122972/article/details/85989131
标签:tensorflow,ckpt,节点名称
0
投稿
猜你喜欢
Python线程threading模块用法详解
2023-01-30 03:55:01
Go语言中的闭包详解
2023-06-30 05:27:14
Sql语句优化注意
2009-10-31 13:15:00
Python查找数组中数值和下标相等的元素示例【二分查找】
2023-11-04 12:55:45
python3 中的字符串(单引号、双引号、三引号)以及字符串与数字的运算
2022-08-26 07:14:52
ASP 80040e14错误的解决方法
2011-03-17 11:21:00
OpenCV实现车辆识别和运动目标检测
2023-09-22 18:55:40
浅析Python中线程以及线程阻塞
2022-03-06 22:14:23
asp中记录集对象的getrows和getstring用法分析
2012-11-30 20:09:49
UTF-8转为GB2312编码的asp函数
2007-08-23 13:42:00
asp生成UTF-8格式的文件方法
2008-01-26 20:59:00
Python Django教程之实现天气应用程序
2022-03-22 23:02:43
如何使用Python发送HTML格式的邮件
2022-10-01 12:24:02
Python locust工具使用详解
2021-11-04 04:04:56
python列表数据增加和删除的具体实例
2021-08-05 15:11:09
Python获取好友地区分布及好友性别分布情况代码详解
2023-05-12 12:59:35
SQL文本字段的数字排序问题
2008-11-18 16:47:00
解剖JavaScript中的null和undefined
2009-03-01 12:49:00
asp显示字符串前5个文字
2008-08-10 18:47:00
Python selenium文件上传下载功能代码实例
2021-04-24 07:21:28