Pytorch中CNN图像回归问题预测值都一样_cnn预测值全是一个

作者：Gausst松鼠会 | 2024-02-19 09:36:19

踩

cnn预测值全是一个

Pytorch中CNN图像回归问题预测值都一样

**
上网也查阅了许多资料，然后对比各种方法都试了一遍，归结为以下几点：
1、出现预测值都一样的情况，一般都是在某一层梯度消失了，然后导致输入到下一层的参数为0，再而后面的值都是一样的。
2、可以添加代码来查看网络训练时的参数变化

List item

 for parameters in net.parameters():
   		 print(parameters)
1
2

3、解决方案：
①在每个卷积层之后添加bn层，也要在linear层和激活层之间添加bn层，目的是对隐藏层输入数据进行归一化。
②初始化网络参数，指的是对卷积层和线性层的操作。
③在forward中先针对卷积层之后的bn层，添加maxpool层和激活函数，再是对linear层之后的bn层添加激活函数。
④注意前面的激活函数都用sigmoid，最后一层用relu。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/113970