SavedModelBundle

public class SavedModelBundle

SavedModelBundle representa um modelo carregado do armazenamento.

O modelo consiste em uma descrição do cálculo (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 como um buffer de protocolo MetaGraphDef .

Classes aninhadas

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

Constantes

Fragmento

DEFAULT_TAG

Métodos Públicos

Mapa <corda, tensor >	chamada (argumentos Map <String, Tensor >) Invoca a função padrão diretamente deste modelo.
vazio	fechar () Libera recursos (o Graph e a Session ) associados ao pacote configurável do modelo salvo.
static SavedModelBundle.Exporter	exportador (String exportDir) Exporte um modelo salvo.
ConcreteFunction	função (String signatureKey) Retorne um ConcreteFunction correspondente à assinatura da função.
Gráfico	gráfico () Retorna o gráfico que descreve o cálculo executado pelo modelo.
estático 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.
MetaGraphDef	metaGraphDef () Retorna o buffer de protocolo MetaGraphDef associado ao modelo salvo.
Sessão	sessão () Retorna a Session com a qual realizar o cálculo usando o modelo.
Lista < Assinatura >	assinaturas () Retorne a assinatura de todas as funções disponíveis neste modelo salvo.

Métodos herdados

Da classe java.lang.Object

boleano	igual a (objeto arg0)
aula final <?>	getClass ()
int	hashCode ()
vazio final	notificar ()
vazio final	notificar tudo ()
Fragmento	toString ()
vazio final	espera (long arg0, int arg1)
vazio final	espera (long arg0)
vazio final	espera ()

Da interface java.lang.AutoCloseable

vazio abstrato

fechar ()

Constantes

public static final String DEFAULT_TAG

Valor constante: "servir"

Métodos Públicos

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

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 é 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 seus nomes de assinatura

Devoluções

• lista de tensores de saída, mapeada pelo nome da assinatura

Lança

Exceção de argumento ilegal	se nenhuma função pode ser selecionada por padrão

public void close ()

Libera recursos (o Graph e a Session ) associados ao pacote configurável do modelo salvo.

public static SavedModelBundle.Exporter exporter (String exportDir)

Exporte um modelo salvo.

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

Parâmetros

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

função public ConcreteFunction (String signatureKey)

Retorne 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

Exceção de argumento ilegal	se a signatureKey não for encontrada neste 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 MetaGraphDef metaGraphDef ()

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

Lista pública < Assinatura > assinaturas ()

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