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
  • asp之家 网络编程 m.aspxhome.com