从训练好的tensorflow模型中打印训练变量实例

yipeiwu_com6年前Python基础

从tensorflow 训练后保存的模型中打印训变量:使用tf.train.NewCheckpointReader()

import tensorflow as tf
reader = tf.train.NewCheckpointReader('path/alexnet/model-330000')
dic = reader.get_variable_to_shape_map()
print dic

打印变量

w = reader.get_tensor("fc1/W")
print type(w)
print w.shape
print w[0]

以上这篇从训练好的tensorflow模型中打印训练变量实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

django实现分页的方法

本文实例讲述了django实现分页的方法。分享给大家供大家参考。具体如下: Python代码如下: #!/usr/bin/env python # -*- coding: utf-8...

Python 遍历子文件和所有子文件夹的代码实例

Python 遍历子文件和所有子文件夹的代码实例

最近看ECShop到网上找资料,发现好多说明ECShop的文件结构不全面,于是想自己弄个出来。但这是个无聊耗时的工作,自己就写了个Python脚本,可以递归遍历目录下的所有文件和所有子目...

每天迁移MySQL历史数据到历史库Python脚本

本文实例为大家分享了Python每天迁移MySQL历史数据到历史库的具体代码,供大家参考,具体内容如下 #!/usr/bin/env python # coding:utf-8...

python的xpath获取div标签内html内容,实现innerhtml功能的方法

python的xpath没有获取div标签内html内容的功能,也就是获取div或a标签中的innerhtml,写了个小程序实现一下: 源代码 [webadmin@centos7 c...

发布你的Python模块详解

我们在学习Python的时候,除了用pip安装一些模块之外,有时候会从网站下载安装包下来安装,我也想要把我自己编写的模块做成这样的安装包,该怎么办,如何发布呢? 大概需要以下四个步骤:...