Pytorch詳解BCELoss和BCEWithLogitsLoss

BCELoss 在圖片多標籤分類時,如果3張圖片分3類,會輸出一個3*3的矩陣。 先用Sigmoid給這些值都搞到0~1之間: 假設Target是: BCELoss是 − 1 n ∑ ( y n × ln ⁡ x n + ( 1 − y n ) × ln ⁡ ( 1 − x n ) ) -\frac 1 n\sum(y_n \times \ln x_n+(1-y_n) \times \ln(1-x
相關文章
相關標籤/搜索