以前我浏览博客的时候记得别人说过,BCELoss与CrossEntropyLoss都是用于分类问题。可以知道,BCELoss是Binary CrossEntropyLoss的缩写,BCELoss CrossEntropyLoss的一个特例,只用于二分类问题,而CrossEntropyLoss可以用于二分类,也可以用于多分类。 不过我重新查阅了一下资料,发现同样是处理二分类问题,BCELoss与CrossEntropyLoss是不同的。下面我详细讲一下哪里不同。 1、使用nn.BCELoss需要在该层前面加上Sigmoid函数。 公式如下: 2、使用nn.CrossEntropyLoss会自动
2022-07-29 11:11:55 47KB op oss ros
1
今天小编就为大家整理了一篇细数nn.BCELoss与nn.CrossEntropyLoss的区别,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
2021-10-12 19:49:25 48KB nn.BCELoss nn.CrossEntropyLoss
1