近期在做人脸表情识别时,遇到了一个问题,就是模型写好进行训练时,出现了不收敛的现象。情况如下:
出现这种问题,很难无从下手,一步一步的排查各个环节,首先检查模型,发现模型并木有问题。部分模型代码:
def train_model():
# 构建模型----------------------------------------------------------
x = tf.placeholder(tf.float32, [None, 128, 128, 1])
y_ = tf.placeholder(tf.int32, [None, ])
y_out, logits = deepnn(x)
loss = loss_value(y_out, y_)
train_step = tf.train.AdamOptimizer(learning_rate=0.0005).minimize(loss)
correct_prediction = tf.equal(tf.cast(tf.argmax(y_out, 1), tf.int32), y_)
accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))
然后检查loss函数,发现这部也没有问题,截取loss函数代码: