Windows下運用Tensorflow object detection API訓練出現的問題--爬坑

剛開始學習Tensorflow object detection API遇到了不少問題,下面就遇到的問題做下記錄。 我是在CPU下訓練的模型,然後遇到訓練到一百多步的時候會報這個錯誤,如圖。用的訓練模型爲ssd_mobilenet_v1_coco。 然後就各種查資料,最後把訓練模型中的 batch_size: 1 改爲 batch_size: 5 成功的把困擾好幾天的問題解決了。
相關文章
相關標籤/搜索