Jeskoo 2021-09-27 20:29
浏览 59
已结题

训练回归模型,输出都在标签的均值附近,

本人正在训练一个回归模型,标签的取值是50~100,主要是75,有部分是50,很少有大于75的,但是训练过程中的输出基本都集中在75附近,训练集有3000+图片验证集有1000+图片,训练集和标签的样子如下:

img

img

下图是一个batch中的标签和训练输出,可以看到图像的标签和模型的输出差别很大,

img

我用的超参数是:
optimizer = Adam,
损失函数 = MSE,
batchsize = 32,
lr = 0.001
网络模型如下:

def get_model():
    model = nn.Sequential(

    nn.Conv2d(3,24,5,stride=2),
    nn.ReLU(),
    nn.MaxPool2d(2, stride=1),

    nn.Conv2d(24,36,5,stride=2),
    nn.ReLU(),
    nn.MaxPool2d(2, stride=1),
    
    nn.Conv2d(36,48,5,stride=2),
    nn.ReLU(),
    nn.MaxPool2d(2, stride=1),
    
    nn.Conv2d(48,64,3,stride=1),
    nn.ReLU(),
    nn.MaxPool2d(2, stride=1),
    
    nn.Dropout2d(p=0.2),
    
    nn.Conv2d(64,64,3,stride=1),
    nn.ReLU(),    
    
    nn.Flatten(),
    
    nn.Dropout2d(p=0.2),
    
    nn.Linear(224000,100),
    nn.ReLU(),
    
    nn.Linear(100,50),
    nn.ReLU(),
    
    nn.Linear(50,10),
    nn.ReLU(),
    
    nn.Linear(10,1),
    )
    return model
model = get_model()

请问有明白的指导一下吗?感激不尽!

  • 写回答

0条回答 默认 最新

    报告相同问题?

    问题事件

    • 系统已结题 10月5日
    • 创建了问题 9月27日

    悬赏问题

    • ¥15 关于#matlab#的问题:在模糊控制器中选出线路信息,在simulink中根据线路信息生成速度时间目标曲线(初速度为20m/s,15秒后减为0的速度时间图像)我想问线路信息是什么
    • ¥15 banner广告展示设置多少时间不怎么会消耗用户价值
    • ¥16 mybatis的代理对象无法通过@Autowired装填
    • ¥15 可见光定位matlab仿真
    • ¥15 arduino 四自由度机械臂
    • ¥15 wordpress 产品图片 GIF 没法显示
    • ¥15 求三国群英传pl国战时间的修改方法
    • ¥15 matlab代码代写,需写出详细代码,代价私
    • ¥15 ROS系统搭建请教(跨境电商用途)
    • ¥15 AIC3204的示例代码有吗,想用AIC3204测量血氧,找不到相关的代码。