pytorch做故障诊断遇到的问题

Question

问题遇到的现象和发生背景

用pytorch做轴承故障检测时，训练结果这样子正常么，直接100%了。

遇到的现象和发生背景，请写出第一个错误信息

没有报错

用代码块功能插入代码，请勿粘贴截图。

使用的是AlexNet模型

运行结果及详细报错内容

1.0371199010705823e-07
1.0728834709539115e-08
0.0
0.0
0.0
0.0
训练两轮就直接降为0了

我的解答思路和尝试过的方法

我把模型改了一下，使用两个简单卷积池化层，激活函数用relu，结果也是这样，直接100

我想要达到的结果，如果你需要快速回答，请尝试 “付费悬赏”

Answer 1

望采纳！！！点击回答右侧采纳即可！！
可能的原因

1. 模型过于简单，没有足够的参数来拟合数据，导致模型过拟合；
2. 数据集不够多，模型没有足够的数据来学习；
3. 数据集中的样本不够多，模型没有足够的样本来学习；
4. 数据集中的样本分布不均匀，模型没有足够的样本来学习；
5. 模型的超参数设置不当，导致模型过拟合。