【发布时间】:2020-03-12 05:57:47
【问题描述】:
我正在尝试使用 ssd_mobilenet_v2_quantized_300x300_coco.config 进行训练
当我尝试训练时它会显示这个
W tensorflow/core/common_runtime/bfc_allocator.cc:239] 分配器 (GPU_0_bfc) 用 freed_by_count=0 分配 927.72MiB 时内存不足。调用者表示这不是失败,但可能意味着如果有更多可用内存可能会提高性能。
这些是我用来训练的命令
python train.py --pipeline_config_path=training/ssd_mobilenet_v2_quantized_300x300_coco.config --train_dir=training/ --alsologtostderr
【问题讨论】:
-
仅供参考,如果我使用 ssd_mobilenet_v2_coco.config 这个问题就不会发生
-
什么GPU?配置说明了什么?
-
@geometrikal GTX 1080 ti
标签: tensorflow