BCELoss和BCEWithLogitsLoss的区别
BCELoss和BCEWithLogitsLoss要求的input都是经过sigmoid产生的分类概率,target是0或1的二分类。假设我们有一个3×3的输入,也就是batch_size是3,target是3×3,表示有3个标签。现在用这个例子做一个演示:import torchimport torch.nn as nnm = nn.Sigmoid()loss1 = nn.BCELoss()loss2 = nn.BCEWithLogitsLoss()input = torch.randn
原创
2022-01-06 20:40:41 ·
696 阅读 ·
0 评论