pytorch学习笔记——BCE与CE_torch bce ce

作者：天景科技苑 | 2024-08-09 03:00:29

踩

torch bce ce

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
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/天景科技苑/article/detail/951112