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 Session ) associados ao pacote de modelo salvo.
Gráfico
gráfico ()
Retorna o gráfico que descreve o cálculo realizado pelo modelo.
estático SavedModelBundle
load (String exportDir, String... tags)
Carregue um modelo salvo de um diretório de exportação.
estático SavedModelBundle.Loader
carregador (String exportDir)
Carregue um modelo salvo.
byte[]
metaGraphDef ()
Retorna o buffer de protocolo MetaGraphDef serializado associado ao modelo salvo.
Sessão
sessão ()
Retorna a Session com a qual realizar a computação usando o modelo.

Métodos Herdados

Métodos públicos

vazio público fechar ()

Libera recursos (o Graph e 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.

carga SavedModelBundle estática pública (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 Saved Model .

Este método é um atalho 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.

carregador estático público 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 de protocolo MetaGraphDef serializado associado ao modelo salvo.

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

Retorna a Session com a qual realizar a computação usando o modelo.

Devoluções
  • a sessão inicializada