尝试使用Tensorflow的对象检测API(https://github.com/tensorflow/models/tree/master/research/object_detection)训练SSD Mobilenet“ ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03”时,我无法充分利用GPU的最大功能。
我正在使用自定义数据集来微调SSD Mobilenet V2,但是在训练时,每个GPU仅利用了约160 MB的内存。我正在使用Nvidia 1080-ti GPU进行培训。我使用的批次大小为24。