SavedModelBundle

общедоступный класс SavedModelBundle

SavedModelBundle представляет модель, загруженную из хранилища.

Модель состоит из описания вычисления ( Graph ), Session с тензорами (например, параметрами или переменными в графике), инициализированными значениями, сохраненными в хранилище, и описания модели в виде буфера протокола MetaGraphDef .

Вложенные классы

сорт SavedModelBundle.Exporter Параметры экспорта SavedModel.
сорт SavedModelBundle.Loader Параметры загрузки SavedModel.

Константы

Нить DEFAULT_TAG

Публичные методы

Карта<Строка, Тензор >
вызов (Аргументы Map<String, Tensor >)
Вызывает функцию по умолчанию непосредственно из этой модели.
пустота
закрывать ()
Освобождает ресурсы ( Graph и Session ), связанные с сохраненным пакетом модели.
статический SavedModelBundle.Exporter
экспортер (String ExportDir)
Экспортируйте сохраненную модель.
КонкретнаяФункция
функция (String SignatureKey)
Верните ConcreteFunction , соответствующую сигнатуре функции.
График
график ()
Возвращает график, описывающий вычисления, выполняемые моделью.
статический SavedModelBundle
загрузить (теги String ExportDir, String...)
Загрузите сохраненную модель из каталога экспорта.
статический SavedModelBundle.Loader
загрузчик (String ExportDir)
Загрузите сохраненную модель.
Метаграфдеф
метаграфдеф ()
Возвращает буфер протокола MetaGraphDef, связанный с сохраненной моделью.
Сессия
сеанс ()
Возвращает Session , с помощью которого выполняются вычисления с использованием модели.
Список < Подпись >
подписи ()
Верните подпись всех функций, доступных в этой сохраненной модели.

Унаследованные методы

Константы

общедоступная статическая финальная строка DEFAULT_TAG

Постоянное значение: «обслуживать».

Публичные методы

общедоступный вызов Map<String, Tensor > (аргументы Map<String, Tensor >)

Вызывает функцию по умолчанию непосредственно из этой модели.

Выбор функции по умолчанию осуществляется на основе первого из следующих условий, которое является истинным:

  • Эта функция является единственной доступной подписью, прикрепленной к основному графику этой сохраненной модели.
  • Функция сопоставляется с именем подписи по умолчанию, то есть «serving_default».

Вызывающая сторона несет ответственность за закрытие всех возвращенных тензоров.

Параметры
аргументы список входных тензоров, сопоставленных с их сигнатурным именем
Возврат
  • список выходных тензоров, сопоставленных с именем сигнатуры
Броски
IllegalArgumentException если ни одна функция не может быть выбрана по умолчанию

public void закрыть ()

Освобождает ресурсы ( Graph и Session ), связанные с сохраненным пакетом модели.

общедоступный статический экспортер SavedModelBundle.Exporter (String ExportDir)

Экспортируйте сохраненную модель.

Возвращает объект Exporter для настройки параметров конфигурации перед фактическим сохранением модели.

Параметры
каталог экспорта путь к каталогу, содержащему сохраненную модель.

публичная функция ConcreteFunction (String SignatureKey)

Верните ConcreteFunction , соответствующую сигнатуре функции.

ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
 Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
 

Параметры
ключ подписи имя SignatureDef в сохраненной модели.
Возврат
  • объект, который можно использовать для вызова функции
Броски
IllegalArgumentException если signatureKey не найдена в этой сохраненной модели.

граф публичного графика ()

Возвращает график, описывающий вычисления, выполняемые моделью.

общедоступная статическая загрузка SavedModelBundle (теги String ExportDir, String...)

Загрузите сохраненную модель из каталога экспорта. Загружаемая модель должна быть создана с помощью API сохраненной модели .

Этот метод является сокращением для:

SavedModelBundle.loader().withTags(tags).load();
 

Параметры
каталог экспорта путь к каталогу, содержащему сохраненную модель.
теги теги, идентифицирующие конкретный метаграфдеф для загрузки.
Возврат
  • пакет, содержащий график и связанный сеанс.

общедоступный статический загрузчик SavedModelBundle.Loader (String ExportDir)

Загрузите сохраненную модель.

Возвращает объект Loader , который может устанавливать параметры конфигурации перед фактической загрузкой модели.

Параметры
каталог экспорта путь к каталогу, содержащему сохраненную модель.

публичный MetaGraphDef метаGraphDef ()

Возвращает буфер протокола MetaGraphDef, связанный с сохраненной моделью.

сеанс публичной сессии ()

Возвращает Session , с помощью которого выполняются вычисления с использованием модели.

Возврат
  • инициализированный сеанс

общедоступный список < Подпись > подписи ()

Верните подпись всех функций, доступных в этой сохраненной модели.