SavedModelBundle

clase pública SavedModelBundle

SavedModelBundle representa un modelo cargado desde el almacenamiento.

El modelo consta de una descripción del cálculo (un Graph ), una Session con tensores (por ejemplo, parámetros o variables en el gráfico) inicializados a valores guardados en el almacenamiento y una descripción del modelo como búfer de protocolo MetaGraphDef .

Clases anidadas

clase SavedModelBundle.Exporter Opciones para exportar un modelo guardado.
clase SavedModelBundle.Loader Opciones para cargar un modelo guardado.

Constantes

Cuerda DEFAULT_TAG

Métodos públicos

Mapa <Cadena, Tensor >
llamar (Map <String, Tensor > argumentos)
Invoca la función predeterminada directamente desde este modelo.
vacío
cerrar ()
Libera recursos (el Graph y la Session ) asociados con el paquete de modelo guardado.
static SavedModelBundle.Exporter
exportador (String exportDir)
Exporta un modelo guardado.
Hormigón Función
función (String signatureKey)
Devuelve una función ConcreteFunction correspondiente a la firma de la función.
Grafico
gráfico ()
Devuelve el gráfico que describe el cálculo realizado por el modelo.
static SavedModelBundle
load (String exportDir, String ... etiquetas)
Cargue un modelo guardado desde un directorio de exportación.
static SavedModelBundle.Loader
cargador (String exportDir)
Cargue un modelo guardado.
MetaGraphDef
metaGraphDef ()
Devuelve el búfer del protocolo MetaGraphDef asociado con el modelo guardado.
Sesión
sesión ()
Devuelve la Session con la que realizar el cálculo utilizando el modelo.
Lista < Firma >
firmas ()
Devuelve la firma de todas las funciones disponibles en este modelo guardado.

Métodos heredados

Constantes

Cadena final estática pública DEFAULT_TAG

Valor constante: "servir"

Métodos públicos

Public Map <String, Tensor > call (Map <String, Tensor > argumentos)

Invoca la función predeterminada directamente desde este modelo.

La selección de la función predeterminada se realiza en función de la primera de las siguientes condiciones que sea verdadera:

  • La función es la única firma disponible adjunta al gráfico principal de este modelo guardado
  • La función está asignada al nombre de la firma predeterminado, que es "servicio_predeterminado".

La persona que llama es responsable de cerrar todos los tensores devueltos.

Parámetros
argumentos lista de tensores de entrada, mapeados por su nombre de firma
Devoluciones
  • lista de tensores de salida, mapeados por el nombre de la firma
Lanza
Argumento de excepción ilegal si no se puede seleccionar ninguna función por defecto

cierre del vacío público ()

Libera recursos (el Graph y la Session ) asociados con el paquete de modelo guardado.

exportador public static SavedModelBundle.Exporter (String exportDir)

Exporta un modelo guardado.

Devuelve un objeto Exporter para establecer las opciones de configuración antes de guardar el modelo.

Parámetros
exportDir la ruta del directorio que contiene un modelo guardado.

función pública ConcreteFunction (String signatureKey)

Devuelve una función ConcreteFunction correspondiente a la firma de la función.

ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
 Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
 

Parámetros
signatureKey nombre de SignatureDef en el modelo guardado.
Devoluciones
  • objeto que se puede utilizar para realizar llamadas a una función
Lanza
Argumento de excepción ilegal si no se encuentra signatureKey en este modelo guardado.

gráfico gráfico público ()

Devuelve el gráfico que describe el cálculo realizado por el modelo.

carga public static SavedModelBundle (String exportDir, String ... tags)

Cargue un modelo guardado desde un directorio de exportación. El modelo que se está cargando debe crearse utilizando la API de modelo guardado .

Este método es una abreviatura de:

SavedModelBundle.loader().withTags(tags).load();
 

Parámetros
exportDir la ruta del directorio que contiene un modelo guardado.
etiquetas las etiquetas que identifican la metagraphdef específica a cargar.
Devoluciones
  • un paquete que contiene el gráfico y la sesión asociada.

cargador public static SavedModelBundle.Loader (String exportDir)

Cargue un modelo guardado.

Devuelve un objeto Loader que puede establecer opciones de configuración antes de cargar el modelo,

Parámetros
exportDir la ruta del directorio que contiene un modelo guardado.

public MetaGraphDef metaGraphDef ()

Devuelve el búfer del protocolo MetaGraphDef asociado con el modelo guardado.

sesión de sesión pública ()

Devuelve la Session con la que realizar el cálculo utilizando el modelo.

Devoluciones
  • la sesión inicializada

Lista pública < Firma > firmas ()

Devuelve la firma de todas las funciones disponibles en este modelo guardado.