SavedModelBundle

public class 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) inicializada 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 a Session ) associados ao pacote configurável do modelo salvo.
Gráfico
gráfico ()
Retorna o gráfico que descreve o cálculo realizado pelo modelo.
static 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.
byte[]
metaGraphDef ()
Retorna o buffer de protocolo MetaGraphDef serializado associado ao modelo salvo.
Sessão
sessão ()
Retorna a Session com a qual realizar o cálculo usando o modelo.

Métodos herdados

Métodos Públicos

public void close ()

Libera recursos (o Graph e a Session ) associados ao pacote configurável do 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 byte [] 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 o cálculo usando o modelo.

Devoluções
  • a sessão inicializada