刚开始学习Tensorflow object detection API遇到了不少问题,下面就遇到的问题做下记录。
我是在CPU下训练的模型,然后遇到训练到一百多步的时候会报这个错误,如图。用的训练模型为ssd_mobilenet_v1_coco。
然后就各种查资料,最后把训练模型中的 batch_size: 1 改为 batch_size: 5 成功的把困扰好几天的问题解决了。