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