網絡訓練中,loss曲線非常奇怪
交叉熵怎麼會有負數。
經過排查,交叉熵不是有個負對數嗎,當網絡輸出的概率是0-1時,正數。可當網絡輸出大於1的數,就有可能變成負數。
所以加上一行就行了
out1 = F.softmax(out1, dim=1)
網絡訓練中,loss曲線非常奇怪
交叉熵怎麼會有負數。
經過排查,交叉熵不是有個負對數嗎,當網絡輸出的概率是0-1時,正數。可當網絡輸出大於1的數,就有可能變成負數。
所以加上一行就行了
out1 = F.softmax(out1, dim=1)