1. 程式人生 > >tensorflow下怎麼解決語義分割交叉熵損失總是nan

tensorflow下怎麼解決語義分割交叉熵損失總是nan

本次訓練一個帶有語義分割任務的網路,發現語義分割的交叉熵損失一直是nan,而且是從剛開始迭代就是nan,檢查了網路一直沒發現問題,學習率調小也不起作用,推測是損失函式計算中log傳入了0,考慮到用的啟用函式是tanh,會產生0值,所以考慮將啟用函式換成不帶0的sigmoid函式,換完之後發現立馬就是正常損失值開始迭代了