2017-07-16 111 views
0

我是Tensorflow的新手,所以这个问题可能真的很愚蠢。Tensorflow Layers API CNN参数在训练过程中不会改变

我一直在尝试使用Tensorflow为MNIST手写数字数据集编写一个简单的CNN。问题在于参数未被优化器更新(由Tensorboard摘要监控)。
尽管由Layers API创建的范围看起来很奇怪,但图似乎还行。渐变是从每个图层计算出来的。 请帮忙!

我使用从这里训练数据:http://yann.lecun.com/exdb/mnist/

下面是代码

import tensorflow as tf 

DATA = 'train-images.idx3-ubyte' 
LABELS = 'train-labels.idx1-ubyte' 
NUM_EPOCHS = 2 
BATCH_SIZE = 15 
#Data definition 
data_queue = tf.train.string_input_producer([DATA,]) 
label_queue = tf.train.string_input_producer([LABELS,]) 

reader_data = tf.FixedLengthRecordReader(record_bytes=28*28, header_bytes = 16) 
reader_labels = tf.FixedLengthRecordReader(record_bytes=1, header_bytes = 8) 

(_,data_rec) = reader_data.read(data_queue) 
(_,label_rec) = reader_labels.read(label_queue) 

image = tf.decode_raw(data_rec, tf.uint8) 
image = tf.reshape(image, [28, 28, 1]) 
label = tf.decode_raw(label_rec, tf.uint8) 
label = tf.reshape(label, [1]) 

image_batch, label_batch = tf.train.shuffle_batch([image, label], 
               batch_size=BATCH_SIZE, 
               capacity=100, 
               min_after_dequeue = 30) 
#Layers definition 
conv = tf.layers.conv2d(
    inputs=tf.cast(image_batch, tf.float32), 
    filters=15, 
    kernel_size=[5,5], 
    padding='same', 
    activation=tf.nn.relu) 

conv1 = tf.layers.conv2d(
    inputs=conv, 
    filters=15, 
    kernel_size=[3,3], 
    padding='same', 
    activation=tf.nn.relu) 

pool_flat = tf.reshape(conv1, [BATCH_SIZE, -1]) 

dense1 = tf.layers.dense(inputs=pool_flat, units=30, activation=tf.nn.relu) 

output = tf.nn.softmax(tf.layers.dense(inputs=dense1, units=10)) 

#train operation definition 
onehot_labels = tf.one_hot(indices=tf.cast(tf.reshape(label_batch,[-1]), tf.int32), depth=10) 

loss = tf.losses.softmax_cross_entropy(onehot_labels=onehot_labels, 
             logits=output) 

global_step = tf.Variable(0,name='global_step',trainable=False) 
train_op = tf.train.GradientDescentOptimizer(0.05).minimize(loss, global_step = global_step) 

#Summaries definition 

for var in tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='conv2d'): 
    tf.summary.histogram(var.name, var) 
for var in tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='conv2d_1'): 
    tf.summary.histogram(var.name, var) 
for var in tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='dense'): 
    tf.summary.histogram(var.name, var) 
for var in tf.get_collection(tf.GraphKeys.GLOBAL_VARIABLES, scope='dense_1'): 
    tf.summary.histogram(var.name, var) 
tf.summary.image("inp", image_batch, max_outputs =1) 
loss_summary = tf.summary.scalar("loss", loss) 
summaries = tf.summary.merge_all() 

#init 
sess = tf.Session() 
summary_writer = tf.summary.FileWriter('log_simple_stats', sess.graph) 
coord = tf.train.Coordinator() 
threads = tf.train.start_queue_runners(coord=coord, sess=sess) 
sess.run(tf.global_variables_initializer()) 

#loop 
for i in range((60000*NUM_EPOCHS)//BATCH_SIZE): 
    sess.run(train_op) 
    if(i%100): 
     merged = sess.run(summaries) 
     summary_writer.add_summary(merged, i) 


coord.request_stop() 
coord.join(threads) 

编辑 定制层给予相同的结果。

自定义层:

def convol(input, inp, outp, name="conv"): 
    with tf.name_scope(name): 
     w = tf.Variable(tf.truncated_normal([5, 5, inp, outp], stddev=0.1),name="W") 
     b = tf.Variable(tf.constant(0.1, shape=[outp]), name="B") 
     filtered = tf.nn.conv2d(input, w, strides=[1,1,1,1], padding="SAME", name="conv2d") 
     activation = tf.nn.relu(features=(filtered+b), name="activation") 
     tf.summary.histogram(name=w.name, values=w) 
     tf.summary.histogram(name=b.name, values=b) 
     tf.summary.histogram(name=activation.name, values=activation) 
     return activation 

def dense(input, inp, outp, name="dense"): 
    with tf.name_scope(name): 
     w = tf.Variable(tf.truncated_normal([inp, outp], stddev=0.1), name="W") 
     b = tf.Variable(tf.constant(0.1, shape=[outp]), name="B") 
     act = tf.matmul(input, w) + b 
     tf.summary.histogram(name=w.name, values=w) 
     tf.summary.histogram(name=b.name, values=b) 
     tf.summary.histogram(name="activation", values=act) 
     return act 

编辑:

所以一段时间后,这个和MNIST例子搞乱从TF我注意到,权重不被教训。我处理数据读取的方式已经搞乱了梯度计算。我刚刚录制了将MNIST数据集读入我的代码的类,并且它无需调整参数即可100%运行。

回答

0

我有同样的问题,我convnet和我的原因只不过是以下组合:

  • 没有足够的迭代:它可能需要相当长一段时间的变化变得可见(几十数千次迭代)
  • 模型复杂化:大幅减少第一次测试的过滤器数量,然后慢慢增加它们以适合您的用例,以确保它不是别的。

为了获得更好的调试,尝试用Tensorboard可视化你的过滤器,这个要点对我帮助很大:

https://gist.github.com/kukuruza/03731dc494603ceab0c5

无论你的方法(tf.layers &手动创建的变量)应该正确地连接到train_op,所以我不认为有什么错。

+0

感谢您的回答!我会尝试用小参数运行它,但我怀疑事实并非如此。这里有1.8个时代的数据(100000次运行)传递给train_op,它没有改变任何内容。 [graph_from_tensorboard](https://ibb.co/bSt3X5) –

相关问题