Pytorch詳解BCELoss和BCEWithLogitsLoss

BCELoss 在圖片多標籤分類時,如果3張圖片分3類,會輸出一個3*3的矩陣。 先用Sigmoid給這些值都搞到0~1之間: 假設Target是: emmm應該是我上面每次都保留4位小數,算到最後誤差越來越大差了0.0001。不過也很厲害啦哈哈哈哈哈! BCEWithLogitsLoss BCEWithLogitsLoss就是把Sigmoid-BCELoss合成一步。我們直接用剛剛的input驗
相關文章
相關標籤/搜索