tensorflow 训练保存模型4 自由修改模型

上回已经可以得到一个清晰的模型结构,但是还是不够满意,为什么呢,因为预测的时候不需要dropout层,所以想修改接口,直接去除dropout层。 以下方法自己想的,可能有别的更好的方法: 首先,训练的时候得把每层的参数起好名字,否则名字都自动起的,怎么修改网络,直接就懵逼了。 #定义两个pl...

2018-12-19 16:00:26

阅读数 53

评论数 0

tensorflow 训练保存模型3 PB格式

上回说到图看不明白。 所以有了下面的方法: 首先读取刚刚的ckpt文件,保存为pb格式(当然训练的时候直接保存也么有问题) 保存Softmax应该就是把计算Softmax所有必须的变量结构都保存下来,无关的就不要了 import os ckpt_dir = &q...

2018-12-19 14:10:19

阅读数 43

评论数 0

tensorflow 训练保存模型2 ckpt tensorboard

以前写过一篇tensorflow 训练保存模型,本文的改进:换了一种模型,然后采用tensorboard来获取tesnor的名字。 训练模型: import tensorflow as tf from tensorflow.examples.tutorials.mnist import in...

2018-12-19 12:04:40

阅读数 40

评论数 0

tensorboad调试

上文中提到:只有一个模型文件,怎么能知道模型结构呢? 想来想去只有使用tensorboard了。 那么直接使用下列代码: saver = tf.train.import_meta_graph("./Model/ModelSoftmax.ckpt.meta"...

2018-06-22 14:53:57

阅读数 130

评论数 0

tensorflow 训练保存模型

训练模型: import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #mnist已经作为官方的例子,做好了数据下载,分割,转浮点等一系列工作,源码在tensorflow源码中都可以找到 ...

2018-06-22 14:13:13

阅读数 144

评论数 0

tensorflow自己实现SGD功能

手动实现SGD和调用优化器结果比较 import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #mnist已经作为官方的例子,做好了数据下载,分割,转浮点等一系列工作,源码在tensorf...

2018-06-22 11:16:03

阅读数 709

评论数 0

训练cifar10的简单例子

cifar10的数据最好自己先下载好import cifar10,cifar10_input import tensorflow as tf import numpy as np import time#max_steps = 30000 max_steps = 1000 data_dir = '...

2018-05-03 16:53:32

阅读数 273

评论数 1

TensorFlow打印op名称

直接打印.op.name就能打印出名称 .get_shape().as_list()就输出shapedef print_activations(t): print(t.op.name, ' ', t.get_shape().as_list())def inference(images):...

2018-04-08 09:51:08

阅读数 893

评论数 0

tensorflow复习日记(八)保存模型

发现一个问题,原来的代码在同一个文件中保存,读取模型是没有问题的,但是不同文件就不行。 原来是保存分2种,ckpt和pb模式。 ckpt中也保存了图的结构,所以正确用法应该是训练模型中把要使用的变量都起个名字。 然后读取后把这些名字读取出来,作为输入,输出,然后喂数据,进行预测: 保存部分...

2018-03-21 11:35:44

阅读数 72

评论数 0

tensorflow复习日记(七)裁剪LeNet5

统计了上面LeNet5的改进形的参数量:from functools import reduce from operator import muldef get_num_params(): num_params = 0 for variable in tf.trainable_va...

2018-03-16 14:18:43

阅读数 146

评论数 0

tensorflow复习日记(六)LeNet5

终于又到了熟悉的LeNet了,还是直接放代码:import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #mnist已经作为官方的例子,做好了数据下载,分割,转浮点等一系列工作,源码在ten...

2018-03-16 11:23:34

阅读数 90

评论数 0

tensorflow复习日记(五)DropOut

上次发现约复杂的网络,拟合能力越强,但是test和train集的准确率差别变大,说明有过拟合的倾向。 下面是代码:import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #mnist已经...

2018-03-15 20:14:17

阅读数 60

评论数 0

tensorflow复习日记(四)BP

下面直接换个有隐藏层的BP:100个隐藏节点,tanh做激活函数: 代码如下:import tensorflow as tf from tensorflow.examples.tutorials.mnist import input_data #mnist已经作为官方的例子,做好了数据下载,分割...

2018-03-15 19:57:51

阅读数 279

评论数 0

tensorflow复习日记(三)softmax修改版本

1 前面的代码都没有关闭Session,所以:可以采用with tf.Session() as sess:这样的方式,自动关闭Session,或者采用手动关闭2 训练出来的模型没有保存,这样一关闭模型就木有了。 3 一般我们都是按照epoch来设定训练次数的。 所以最终修改的代码:import...

2018-03-15 19:09:34

阅读数 79

评论数 0

tensorflow复习日记(二)softmax模型

http://www.tensorfly.cn/tfdoc/tutorials/mnist_pros.html 网址中的代码比较旧了,在最新的tensorflow上运行有warning和错误。 先上修改后的代码和注释:import tensorflow as tf from tensorflo...

2018-03-15 17:11:30

阅读数 84

评论数 0

tensorflow复习日记(一)线性回归

重新把tensorflow复习一下: http://www.tensorfly.cn/tfdoc/get_started/introduction.html 完整代码+注释:import tensorflow as tf import numpy as np# 使用 NumPy 生成测试数据(...

2018-03-15 10:49:11

阅读数 80

评论数 0

提示
确定要删除当前文章?
取消 删除
关闭
关闭