Высокая загрузка ОЗУ после загрузки модели логического вывода в графический процессор - PullRequest
0 голосов
/ 04 апреля 2019

Обучил архитектуру YOLOv2 на пользовательских изображениях и после замораживания графика. Размер файла Вес модели (.pb) составляет 268 МБ. Как только мы загрузим его в gpu для вывода, он потребляет 7,93 ГБ ОЗУ.Я знаю, что tenorflow распределяет буферы для выходных данных на каждом этапе в начале.Пожалуйста, кто-нибудь может объяснить, почему тензорным потоком используется так много памяти.

...