Есть ли способ предсказать объем памяти, который TensorFlow SavedModel будет требовать для запуска? - PullRequest
0 голосов
/ 12 июня 2019

Я пытаюсь создать API в TensorFlow Java, который может сделать вывод о моделях, использующих ЦП, и есть некоторые модели, которые слишком велики для того, чтобы ЦПУ их запускал. Есть ли способ найти память, которая понадобится модели? Или, если это невозможно, каким-либо образом выделить определенный объем памяти для этой команды, поэтому, когда ей не хватает памяти, программа останавливает выполнение этого метода вместо сбоя.

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