SavedModelBundle представляет модель, загруженную из хранилища.
Модель состоит из описания вычисления ( Graph
), Session
с тензорами (например, параметрами или переменными в графике), инициализированными значениями, сохраненными в хранилище, и описания модели в виде буфера протокола MetaGraphDef .
Вложенные классы
сорт | SavedModelBundle.Exporter | Параметры экспорта SavedModel. | |
сорт | SavedModelBundle.Loader | Параметры загрузки SavedModel. |
Константы
Нить | DEFAULT_TAG |
Публичные методы
Карта<Строка, Тензор > | |
пустота | |
статический SavedModelBundle.Exporter | экспортер (String ExportDir) Экспортируйте сохраненную модель. |
КонкретнаяФункция | функция (String SignatureKey) Верните ConcreteFunction , соответствующую сигнатуре функции. |
График | график () Возвращает график, описывающий вычисления, выполняемые моделью. |
статический SavedModelBundle | загрузить (теги String ExportDir, String...) Загрузите сохраненную модель из каталога экспорта. |
статический SavedModelBundle.Loader | загрузчик (String ExportDir) Загрузите сохраненную модель. |
Метаграфдеф | метаграфдеф () Возвращает буфер протокола MetaGraphDef, связанный с сохраненной моделью. |
Сессия | |
Список < Подпись > | подписи () Верните подпись всех функций, доступных в этой сохраненной модели. |
Унаследованные методы
Константы
общедоступная статическая финальная строка DEFAULT_TAG
Публичные методы
общедоступный вызов Map<String, Tensor > (аргументы Map<String, Tensor >)
Вызывает функцию по умолчанию непосредственно из этой модели.
Выбор функции по умолчанию осуществляется на основе первого из следующих условий, которое является истинным:
- Эта функция является единственной доступной подписью, прикрепленной к основному графику этой сохраненной модели.
- Функция сопоставляется с именем подписи по умолчанию, то есть «serving_default».
Вызывающая сторона несет ответственность за закрытие всех возвращенных тензоров.
Параметры
аргументы | список входных тензоров, сопоставленных с их сигнатурным именем |
---|
Возврат
- список выходных тензоров, сопоставленных с именем сигнатуры
Броски
IllegalArgumentException | если ни одна функция не может быть выбрана по умолчанию |
---|
public void закрыть ()
общедоступный статический экспортер 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
, с помощью которого выполняются вычисления с использованием модели.
Возврат
- инициализированный сеанс
общедоступный список < Подпись > подписи ()
Верните подпись всех функций, доступных в этой сохраненной модели.