[pytorch] 交叉熵損失輸出爲負數

網絡訓練中,loss曲線非常奇怪
在這裏插入圖片描述
交叉熵怎麼會有負數。
經過排查,交叉熵不是有個負對數嗎,當網絡輸出的概率是0-1時,正數。可當網絡輸出大於1的數,就有可能變成負數。
所以加上一行就行了

out1 = F.softmax(out1, dim=1)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章