Pytorch:三个交叉熵损失函数区别

1. torch.nn.CrossEntropyLoss()

用于多分类，本身含有softmax

2. torch.BCELoss()

用于二分类，本身不含有sigmoid，需要在网络里面自己增加

3. torch.BCEWithLogitsLoss()

用于二分类，相当于 BCELoss+sigmoid

posted @ 2021-12-19 21:05  骑猪猪看日出哦  阅读(111)  评论(0)    收藏  举报