當(dāng)網(wǎng)絡(luò)最后一層使用nn.Sigmoid時(shí),就用BCELoss,當(dāng)網(wǎng)絡(luò)最后一層不使用nn.Sigmoid時(shí),就用BCEWithLogitsLoss。 (BCELoss)BCEWithLogitsLoss 用于單標(biāo)簽二分類或者多標(biāo)簽二分類,輸出和目標(biāo)的維度是(batch,C),batch是樣本數(shù)量,C是類別數(shù)量,對(duì)于每一個(gè)batch的C個(gè)值,對(duì)每個(gè)值求sigmoid到0-1之間,所以每個(gè)batch的C個(gè)值之...
www.dbjr.com.cn/article/2121...htm 2025-5-30