在进行UNet实验时,发生一件令人崩溃的事情,将CASIA1.0换成CASIA2.0时,仅仅换了个数据集!仅仅换了个数据集!仅仅换了个数据集!就出现了计算损失一直是0.69
经过多方查找,原因是loss函数的问题,
我在网络输出之后加入了sigmoid激活函数,但是使用的loss函数是BCEWithLogitsLoss。而BCEWithLogitsLoss是BCELoss加上一个sigmoid函数,所以我把loss改为BCELoss之后loss就开始正常波动了!!!!!!!
记录一下,虽然没有什么技术上的提高,,,,遗憾
损失一直是0.69
最新推荐文章于 2024-04-14 17:14:50 发布