问题遇到的现象和发生背景
用pytorch做轴承故障检测时,训练结果这样子正常么,直接100%了。
遇到的现象和发生背景,请写出第一个错误信息
没有报错
用代码块功能插入代码,请勿粘贴截图。
使用的是AlexNet模型
运行结果及详细报错内容
1.0371199010705823e-07
1.0728834709539115e-08
0.0
0.0
0.0
0.0
训练两轮就直接降为0了
我的解答思路和尝试过的方法
我把模型改了一下,使用两个简单卷积池化层,激活函数用relu,结果也是这样,直接100
我想要达到的结果,如果你需要快速回答,请尝试 “付费悬赏”
望采纳!!!点击回答右侧采纳即可!!
可能的原因
- 模型过于简单,没有足够的参数来拟合数据,导致模型过拟合;
- 数据集不够多,模型没有足够的数据来学习;
- 数据集中的样本不够多,模型没有足够的样本来学习;
- 数据集中的样本分布不均匀,模型没有足够的样本来学习;
- 模型的超参数设置不当,导致模型过拟合。