訓練mask-rcnn時報錯,RuntimeWarning: overflow encountered in exp

在py-faster-rcnn的基礎上修改爲mask-rcnn的過程中,遇到很多問題,其中訓練的時候報exp的溢出錯誤。 該錯誤是因爲mask分支的loss過大,在同faster-rcnn共享同一個學習率的時候,梯度回傳過大,導致梯度爆炸,進而出現溢出錯誤。 在faster-rcnn的bbox_transform.py第48行就容易出現該類錯誤。 解決辦法是降低mask分支的loss佔比,將其權重
相關文章
相關標籤/搜索