之前在用yolo v3訓練自己的數據集的時候,會出現loss=nan的情況。這邊給出一點解決方法。
1.查看是否為代碼問題,在計算損失時是否出現負數,分母為0等情況。
2.檢查數據集文件是否標識正確。
3.每一次batch,打印一次loss,檢查是否出現梯度爆炸的情況。若有loss=inf,則可以設定一個閾值。
4.調整參數,我就是因為學習率過小出現了nan值,調大一點就正常了
之前在用yolo v3訓練自己的數據集的時候,會出現loss=nan的情況。這邊給出一點解決方法。
1.查看是否為代碼問題,在計算損失時是否出現負數,分母為0等情況。
2.檢查數據集文件是否標識正確。
3.每一次batch,打印一次loss,檢查是否出現梯度爆炸的情況。若有loss=inf,則可以設定一個閾值。
4.調整參數,我就是因為學習率過小出現了nan值,調大一點就正常了
本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。