Использование графического процессора - обучение SSD Mobil enet (API обнаружения объектов) - PullRequest
0 голосов
/ 10 июля 2020

При попытке обучить SSD Mobil enet "ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03" с помощью API обнаружения объектов Tensorflow (https://github.com/tensorflow/models/tree/master/research/object_detection), я не могу использовать графические процессоры на их максимальной мощности.

Я использую специальный набор данных для точной настройки SSD Mobil enet V2, но во время обучения из каждого графического процессора используется только ~ 160 МБ. Я использую графический процессор Nvidia 1080-ti для обучения. Я использую размер пакета 24.

Вывод nvidia-smi

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...