使用自定义损失函数训练unet时,参数没有更新


epoch=0和epoch=1训练的参数都一样,这个要怎么解决呀?unet应该没有问题,使用BCELoss训练结果正常。
你应该说你的自定义损失函数有问题,然后上代码。参数没有更新:1、学习率为0,2、训练时设置了model.eval(),3、loss函数有误,导致weight的梯度为0
BCE损失函数(Binary Cross-Entropy Loss)是交叉熵损失函数(Cross-Entropy Loss)的一种特例,BCE Loss只应用在二分类任务中。具体的可以看我的关于unet的一个笔记:
自定义损失有问题,大概是定义错方向了。