python - INFO:tensorflow:Error 报告给 Coordinator：,

Question

我正在运行 tensorflow 对象检测 train.py 我正在使用 tensorflow-gpu 1.5、Windows 10、CUDA 9.0、CUDNN 7.0、Python 3.5 和 3* GTX 1070 8GB 内存显卡。

当我运行时python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config，我收到以下错误：

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

甚至在训练开始之前

Answer 1

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