SavedModelBundle

public class SavedModelBundle

SavedModelBundle representa um modelo carregado do armazenamento.

O modelo consiste em uma descrição do cálculo (um Graph ), uma Session com tensores (por exemplo, parâmetros ou variáveis ​​no gráfico) inicializada para valores salvos no armazenamento e uma descrição do modelo como um buffer de protocolo MetaGraphDef .

Classes aninhadas

aula SavedModelBundle.Exporter Opções para exportar um SavedModel.
aula SavedModelBundle.Loader Opções para carregar um SavedModel.

Constantes

Fragmento DEFAULT_TAG

Métodos Públicos

Mapa <corda, tensor >
chamada (argumentos Map <String, Tensor >)
Invoca a função padrão diretamente deste modelo.
vazio
fechar ()
Libera recursos (o Graph e a Session ) associados ao pacote configurável do modelo salvo.
static SavedModelBundle.Exporter
exportador (String exportDir)
Exporte um modelo salvo.
ConcreteFunction
função (String signatureKey)
Retorne um ConcreteFunction correspondente à assinatura da função.
Gráfico
gráfico ()
Retorna o gráfico que descreve o cálculo executado pelo modelo.
estático SavedModelBundle
carregar (String exportDir, String ... tags)
Carregue um modelo salvo de um diretório de exportação.
static SavedModelBundle.Loader
carregador (String exportDir)
Carregue um modelo salvo.
MetaGraphDef
metaGraphDef ()
Retorna o buffer de protocolo MetaGraphDef associado ao modelo salvo.
Sessão
sessão ()
Retorna a Session com a qual realizar o cálculo usando o modelo.
Lista < Assinatura >
assinaturas ()
Retorne a assinatura de todas as funções disponíveis neste modelo salvo.

Métodos herdados

Constantes

public static final String DEFAULT_TAG

Valor constante: "servir"

Métodos Públicos

chamada pública Map <String, Tensor > (argumentos Map <String, Tensor >)

Invoca a função padrão diretamente deste modelo.

A seleção da função padrão é feita com base na primeira das seguintes condições, que é verdadeira:

  • A função é a única assinatura disponível anexada ao gráfico principal deste modelo salvo
  • A função é mapeada para o nome da assinatura padrão, que é "serving_default"

O chamador é responsável por fechar todos os tensores retornados.

Parâmetros
argumentos lista de tensores de entrada, mapeados por seus nomes de assinatura
Devoluções
  • lista de tensores de saída, mapeada pelo nome da assinatura
Lança
Exceção de argumento ilegal se nenhuma função pode ser selecionada por padrão

public void close ()

Libera recursos (o Graph e a Session ) associados ao pacote configurável do modelo salvo.

public static SavedModelBundle.Exporter exporter (String exportDir)

Exporte um modelo salvo.

Retorna um objeto Exporter para definir opções de configuração antes de realmente salvar o modelo.

Parâmetros
exportDir o caminho do diretório que contém um modelo salvo.

função public ConcreteFunction (String signatureKey)

Retorne um ConcreteFunction correspondente à assinatura da função. 

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

Parâmetros
assinaturaKey nome do SignatureDef no modelo salvo.
Devoluções
  • objeto que pode ser usado para fazer chamadas para uma função
Lança
Exceção de argumento ilegal se a signatureKey não for encontrada neste modelo salvo.

gráfico gráfico público ()

Retorna o gráfico que descreve o cálculo executado pelo modelo.

public static SavedModelBundle load (String exportDir, String ... tags)

Carregue um modelo salvo de um diretório de exportação. O modelo que está sendo carregado deve ser criado usando a API de modelo salvo .

Este método é uma abreviação para: 

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

Parâmetros
exportDir o caminho do diretório que contém um modelo salvo.
Tag as tags que identificam o metagraphdef específico a ser carregado.
Devoluções
  • um pacote contendo o gráfico e a sessão associada.

public static SavedModelBundle.Loader loader (String exportDir)

Carregue um modelo salvo.

Retorna um objeto Loader que pode definir opções de configuração antes de realmente carregar o modelo,

Parâmetros
exportDir o caminho do diretório que contém um modelo salvo.

public MetaGraphDef metaGraphDef ()

Retorna o buffer de protocolo MetaGraphDef associado ao modelo salvo.

sessão de sessão pública ()

Retorna a Session com a qual realizar o cálculo usando o modelo.

Devoluções
  • a sessão inicializada

Lista pública < Assinatura > assinaturas ()

Retorne a assinatura de todas as funções disponíveis neste modelo salvo.