我正在运行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'>
甚至在培训开始之前
答案 0 :(得分:0)
尝试调整train.py文件中的batch_size来解决内存问题。