我扩充了我的数据集,并使用Tensorflow对象检测API https://github.com/tensorflow/models/tree/master/research/object_detection在SSD Mobilenet v1 coco上对其进行了训练。
我最初遇到此错误:
ResourceExhaustedError(请参阅上面的回溯):分配时为OOM 形状为[] 24,256,38,38]的张量,类型为float on / job:localhost / replica:0 / task:0 / device:GPU:0通过分配器GPU_0_bfc
然后我将批次大小从24更改为3,并开始训练。
然后我开始在同一增强数据集上使用Faster RCNN v2 Inception进行训练,并得到以下错误:
InvalideArgumentError(请参阅上面的回溯):LossTensor为inf或 南:张量具有NaN值
然后我将批量大小从1更改为32,并得到此错误:
InvalidArgumentError(请参阅上面的回溯):断言失败: [最大框坐标值大于1.100000:]
该如何解决?