GPU使用率-SSD Mobilenet培训(对象检测API)

时间:2020-07-10 11:46:12

标签: python tensorflow object-detection nvidia object-detection-api

尝试使用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。

Output of nvidia-smi

0 个答案:

没有答案