pytorch做故障诊断遇到的问题

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

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

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

没有报错

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

使用的是AlexNet模型

运行结果及详细报错内容

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

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

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

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

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

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