
在进行UNet实验时,发生一件令人崩溃的事情,将CASIA1.0换成CASIA2.0时,仅仅换了个数据集!仅仅换了个数据集!仅仅换了个数据集!就出现了计算损失一直是0.69
经过多方查找,原因是loss函数的问题,
我在网络输出之后加入了sigmoid激活函数,但是使用的loss函数是BCEWithLogitsLoss。而BCEWithLogitsLoss是BCELoss加上一个sigmoid函数,所以我把loss改为BCELoss之后loss就开始正常波动了!!!!!!!
记录一下,虽然没有什么技术上的提高,,,,遗憾
95939869
《损失一直是0.69,损失6666》来自互联网同行内容,若有侵权,请联系我们删除!
来电咨询