在pytorch中使用自己定义的损失函数后,ResNet输出为None
经检测发现是因为在技术loss中有除法。而除数为(A-B),因为A和B在某些数据对下相等所以除数为0,虽然没报错但是输出是None.
解决方法:将除数改为(A-B+1e-6)
在pytorch中使用自己定义的损失函数后,ResNet输出为None
经检测发现是因为在技术loss中有除法。而除数为(A-B),因为A和B在某些数据对下相等所以除数为0,虽然没报错但是输出是None.
解决方法:将除数改为(A-B+1e-6)