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