pytorch 训练编解码网络进行图像重建时mse loss不下降

将transform.normalize()的参数修改为mean = [0.5], std = [0.5]
输入图像为灰度图像,所以只有单通道,修改前loss降至0.7左右不再下降,修改后可以降至0.07
之前试过修改激活函数、优化器等都不能解决问题
但为何如此,还没有搞明白。解决了再来更新

pytorch 训练编解码网络进行图像重建时mse loss不下降最先出现在Python成神之路

版权声明:
作者:admin
链接:https://www.techfm.club/p/17691.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>