您的位置:首页 > 移动开发 > Objective-C

【TensorFlow】使用Object Detection API 训练自己的数据集报错

2018-09-29 09:11 627 查看
错误1:  训练正常开始后,能正常看到日志输出,但中途报错 ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[1435,9,256] .....................


ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[1435,9,256] .....................
个人解决办法:


  修改 模型配置文件中的 batch_size原本是24,我直接调到4后训练运行正常了


总结原因:


  ① 使用的CPU训练的,电脑性能差。 


内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: