Pytorch交叉熵損失函數CrossEntropyLoss報錯解決辦法

在pytorch的官方論壇裏有一我的也遇到了一樣的問題,他把nn.CrossEntropyLoss()換成了nn.MultiLabelSoftMarginLoss()就再也不報錯了。可是通過實驗發現,這個損失函數的效果很是差,遠不如MSELoss。web 最終,我找到了一篇運用交叉熵損失函數的多分類代碼一步步檢查發現了報錯的緣由:svg 在多分類問題中,當損失函數爲nn.CrossEntropyL
相關文章
相關標籤/搜索