我正在尝试使用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
答案 0 :(得分:0)
答案是获得16GB内存GPU。