dog-qiuqiu / Yolo-Fastest

:zap: Based on yolo's ultra-lightweight universal target detection algorithm, the calculation amount is only 250mflops, the ncnn model size is only 666kb, the Raspberry Pi 3b can run up to 15fps+, and the mobile terminal can run up to 178fps+
Other
2k stars 426 forks source link

ej0 m6 #65

Open akioolin opened 3 years ago

akioolin commented 3 years ago

目前嘗試不使用預設的detector 權重開始,訓練自建的資料集。結果會遇到nan,特別是在第二段的v3 loss輸出,會觀察到某一段時間後,loss值直接飆上去,然後就爆掉了。連帶影響到第一段v3 loss變化。

請問各位有遇過類似的狀況嘛?

訓練機器偏小,弱弱的一張1050。跑ubuntu 20.04。使用darknet進行訓練yolo-fastest。自訂的訓練集,有50類,每類約有3000個框框。