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