用nvidia-docker跑深度學習模型

背景 最近實驗室要參加一個目標檢測的比賽,這段時間一直在跑ssd模型,最開始根據作者給的文檔成功編譯後,可以在VOC數據集上進行訓練。由於要用比賽官方的數據集,因此做了幾天的數據集,然後拿自己的數據集訓練的時候,出現了以下報錯:Check failed: a <= b (0 vs. -1.192093-07) 去網上搜了相關的解決方法,全都是說把math_functions.cpp第250行註釋掉
相關文章
相關標籤/搜索