tensorflow查看ckpt各节点名称实例

yipeiwu_com7年前Python基础

运行下列脚本,可以打印出模型各个节点变量的名称:

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为自己的模型路径

以上这篇tensorflow查看ckpt各节点名称实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

Python如何优雅获取本机IP方法

见过很多获取服务器本地IP的代码,个人觉得都不是很好,例如以下这些 不推荐:靠猜测去获取本地IP方法 #!/usr/bin/env python # -*- coding: utf-...

python操作字典类型的常用方法(推荐)

has_key()方法可以检查字典中是否含有指定的键,如果有则返回True,否则就返回False。 语法格式: dictionary_name.has_key(key) dict...

python清除字符串里非字母字符的方法

本文实例讲述了python清除字符串里非字母字符的方法。分享给大家供大家参考。具体如下: s = "hello world! how are you? 0" # Short...

在Django的模型中执行原始SQL查询的方法

有时候你会发现Django数据库API带给你的也只有这么多,那你可以为你的数据库写一些自定义SQL查询。 你可以通过导入django.db.connection对像来轻松实现,它代表当前...

django如何通过类视图使用装饰器

需求:当我们想禁止ip黑名单访问我们的某些页面时,例如注册页面。应该怎么操作呢? 解决方案:这时候我们可以设计一个装饰器,过滤ip黑名单。 装饰器的写法如下: from functo...