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 (uma representação serializada de um buffer de protocolo MetaGraphDef ).

Classes aninhadas

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

Métodos Públicos

vazio
fechar ()
Libera recursos (o Graph e o Session ) associados ao pacote de modelo salvo.
Gráfico
gráfico ()
Retorna o gráfico que descreve o cálculo realizado pelo modelo.
SavedModelBundle estático
carregar (string exportDir, String... tags)
Carregue um modelo salvo de um diretório de exportação.
SavedModelBundle.Loader estático
carregador (String exportDir)
Carregue um modelo salvo.
byte[]
metaGraphDef ()
Retorna o buffer do protocolo MetaGraphDef serializado associado ao modelo salvo.
Sessão
sessão ()
Retorna a Session com a qual será realizada a computação usando o modelo.

Métodos herdados

Métodos Públicos

fechamento de vazio público ()

Libera recursos (o Graph e o Session ) associados ao pacote de 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 abreviação 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.

byte público[] metaGraphDef ()

Retorna o buffer do protocolo MetaGraphDef serializado 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