问题描述:计算BCE Loss
使用pytorch接口
self.bce_loss = nn.BCELoss()
self.bce_loss(pred_cls, tcls)
问题原因:pred_cls没有归一化的0到1之间。
解决办法:
self.bce_loss(torch.sigmoid(pred_cls), tcls)
问题描述:计算BCE Loss
使用pytorch接口
self.bce_loss = nn.BCELoss()
self.bce_loss(pred_cls, tcls)
问题原因:pred_cls没有归一化的0到1之间。
解决办法:
self.bce_loss(torch.sigmoid(pred_cls), tcls)