深度学习问题求大神解答
最近在做基于alexnet网络训练cifar10的分类问题,
现在遇到问题是训练中loss一直保持在2.3左右不变,准确率也不上升
求各位大佬帮我看下代码,指出问题,谢各位大佬
代码如下:
-- coding=UTF-8 --
import tensorflow as tf
import os
import numpy as np
####### 定义网络超参数
global_step = tf.Variable(0, trainable=False)
learning_rate = tf.train.exponential_decay(0.001, global_step, 10000, 0.9, staircase=True) # 学习率
momentum = 0.9 # 动量
training_iters = 70000 # 训练次数
batch_size = 64 # 批次量
####### 定义网络参数
n_classes = 10 # 标签的维度
dropout = 0.5 # Dropout 的概率
####### 占位符输入
x = tf.placeholder(“float”, [None, 32, 32, 3])
y = tf.placeholder(“float”, [None, n_classes])
keep_prob = tf.placeholder(tf.float32)
####### 得到训练集图像地址集合和其标签集合
def get_files(filename):
class_train = []
label_train = []
for train_class in os.listdir(filename):
for pic in os.listdir(filename+train_class):
class_train.append(filename+train_class+’/’+pic)
label_train.append(train_class)
temp = np.array([class_train, label_train])
temp = temp.transpose()
# shuffle the samples
np.random.shuffle(temp)
# after transpose, images is in dimension 0 and label in dimension 1
image_list = list(temp[:, 0])
label_list = list(temp[:, 1])
label_list = [int(i) for i in label_list]
return image_list, label_list<