BCELoss的话只需要网络输出一个通道,CE Loss(Cross Entropy Loss)需要输出n_class个通道。
对于二分类任务可以使用CE Loss输出两个通道,也可以使用BCE Loss输出一个通道。
https://www.jianshu.com/p/5b01705368bb
https://zhuanlan.zhihu.com/p/372628727
https://zhuanlan.zhihu.com/p/580367698
08-20
3091
![](https://csdnimg.cn/release/blogv2/dist/pc/img/readCountWhite.png)