SavedModelBundle

classe pública SavedModelBundle

SavedModelBundle representa um modelo carregado do armazenamento.

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

Classes aninhadas

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

Constantes

Corda DEFAULT_TAG

Métodos Públicos

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

Métodos herdados

Constantes

String final estática pública DEFAULT_TAG

Valor constante: "servir"

Métodos Públicos

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

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 for 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 seu nome de assinatura
Devoluções
  • lista de tensores de saída, mapeados pelo nome da assinatura
Lança
IllegalArgumentException se nenhuma função puder ser selecionada por padrão

fechamento de vazio público ()

Libera recursos (o Graph e o Session ) associados ao pacote de modelo salvo.

exportador estático público SavedModelBundle.Exporter (String exportDir)

Exporte um modelo salvo.

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

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

função pública ConcreteFunction (String subscriptionKey)

Retorna 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
IllegalArgumentException se signatureKey não for encontrada neste modelo salvo.

gráfico gráfico público ()

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

carregamento público estático de SavedModelBundle (tags String exportDir, String...)

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

Este método é uma abreviatura para:

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

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

Carregador público estático SavedModelBundle.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.

público MetaGraphDef metaGraphDef ()

Retorna o buffer do protocolo MetaGraphDef associado ao modelo salvo.

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

Retorna a Session com a qual será realizada a computação usando o modelo.

Devoluções
  • a sessão inicializada

Lista pública< Assinatura > assinaturas ()

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