Pytorch之CrossEntropyLoss() 與 NLLLoss() 的區別

• （三）PyTorch學習筆記——softmax和log_softmax的區別、CrossEntropyLoss() 與 NLLLoss() 的區別、log似然代價函數
• pytorch loss function 總結

 

• NLLLoss 的 輸入 是一個對數概率向量和一個目標標簽(不需要是one-hot編碼形式的). 它不會為我們計算對數概率. 適合網絡的最后一層是log_softmax. 損失函數 nn.CrossEntropyLoss() 與 NLLLoss() 相同, 唯一的不同是它為我們去做 softmax.

• 幾種分割loss

• Pytorch - Cross Entropy Loss