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
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
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1040919.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!