深度学习中学习率是一个很重要的环节,之前跑了好几次模型数据都没有发生变化,仔细研究之后是学习率调的太低了,这里需要调大学习率才能够看到数据相应的变化。
后来进一步进行错误排查之后,发现原因在于使用了BCEWithLogitsloss这一损失函数,这个损失函数相当于sigmoid加上Logitsloss,如果使用这一损失函数前面就不需要加上sigmoid层了,而我在前面的数值之中加上了sigmoid层,相当于有两个sigmoid层,所以训练出来的数据并没有发生明显的变化。
深度学习跑了好几次数据都没有变化
最新推荐文章于 2022-11-14 11:01:03 发布