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 | |
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 |
Métodos herdados
Métodos Públicos
public void close ()
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 objetoLoader
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.