【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] .....................
修改 模型配置文件中的 batch_size原本是24,我直接调到4后训练运行正常了
总结原因:
① 使用的CPU训练的,电脑性能差。
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[1435,9,256] .....................个人解决办法:
修改 模型配置文件中的 batch_size原本是24,我直接调到4后训练运行正常了
总结原因:
① 使用的CPU训练的,电脑性能差。
相关文章推荐
- Tensorflow Object Detection API训练自己的数据集
- (更新视频教程)Tensorflow object detection API 搭建属于自己的物体识别模型(2)——训练并使用自己的模型
- 利用TensorFlow Object Detection API 训练自己的数据集
- 配置TensorFlow的objetc_detection api,训练自己的数据
- 使用tensorflow训练自己的数据
- Windows下安装TensorFlow Object Detection API,训练自己的数据集
- [Tensorflow][Train]Tensorflow 使用Detect API 训练自己的data(local)
- Tensorflow训练自己的Object Detection模型并进行目标检测
- 目标检测(Google object_detection) API 上训练自己的数据集
- 利用TF_Object_detection_API 训练自己的模型
- 使用Tensorflow训练自己的分割数据
- tensorflow object_detection 用自己的数据训练目标检测模型Mobilenet
- Tensorflow物体检测(Object Detection)API的使用
- tensorflow object detection api训练自己的数据集
- 使用Tensorflow训练自己的数据
- 利用TF_Object_detection_API 训练自己的模型
- TensorFlow Object Detection API教程——利用自己制作的数据集进行训练预测和测试
- Tensorflow物体检测(Object Detection)API的使用
- tensorflow学习(1)——训练自己的数据集并进行物体检测(object detection)
- Windows下安装TensorFlow Object Detection API,训练自己的数据集