Как уменьшить использование оперативной памяти для TF Objection Detection API - PullRequest
0 голосов
/ 02 декабря 2018

Я пытаюсь обучить модель твердотельного накопителя, используя API-интерфейс обнаружения объектов tenorflow, в пользовательском наборе данных.У меня есть готовые tfrecords, и обучение начинается, но меня продолжают убивать, потому что у меня недостаточно оперативной памяти.У меня только 8 ГБ оперативной памяти.Я обнаружил, что изменение batch_size, но это только меняет использование моего графического процессора.Есть ли способ уменьшить использование оперативной памяти?В настоящее время я использую train.py из tenorflow / models / research / object_detection / legacy /.

...