binary_cross_entropy需要input和target都在0-1之间,请检查下输入吧。
可以安装上面注释的那一行看看输入值的范围符不符合要求!
说明:有些损失函数对输入有要求体现在,输入值的范围不对会提示,cuda device triggered之类的错误,但是现在没有提示错误,而是loss<0,并且越来越小,训练了200个周期,最后这个损失值变成了负几万的数值。
另外我代码没有自动同步到服务器上,造成了本地代码没有问题,但是在服务器上运行的时候错误没有修复的问题,也要注意一下。