深度学习十分类loss一直保持在2.3是什么问题,求解答

本文探讨了深度学习模型在进行十分类任务时遇到的问题,即损失(loss)始终停留在2.3左右。通过分析网络结构、学习率、动量等超参数,以及数据预处理和批次大小,帮助读者理解可能的原因并提供解决方案。
摘要由CSDN通过智能技术生成

深度学习问题求大神解答


最近在做基于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<

评论 24
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值