INFO:tensorflow:向协调员报告错误:<class'memoryerror'=“”>,

时间:2018-07-24 10:55:36

标签: python tensorflow object-detection

我正在运行tensorflow对象检测train.py 我正在使用带有3 * GTX 1070 8GB内存图形卡的tensorflow-gpu 1.5,Windows 10,CUDA 9.0,CUDN 7.0,Python 3.5。

运行python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config时,出现以下错误:

INFO:tensorflow:Error reported to Coordinator: <class 'MemoryError'>

甚至在培训开始之前

1 个答案:

答案 0 :(得分:0)

尝试调整train.py文件中的batch_size来解决内存问题。