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 (p. ej., parámetros o variables en el gráfico) inicializados con valores guardados en el almacenamiento y una descripción del modelo como un búfer de protocolo MetaGraphDef .

Clases anidadas

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

Constantes

Cadena DEFAULT_TAG

Métodos públicos

Mapa<Cadena, Tensor >
llamar (Mapa<Cadena, Tensor > argumentos)
Invoca la función predeterminada directamente desde este modelo.
vacío
cerca ()
Libera recursos (el Graph y Session ) asociados con el paquete de modelos guardado.
estático SavedModelBundle.Exporter
exportador (cadena exportDir)
Exportar un modelo guardado.
Función Concreta
función (Clave de firma de cadena)
Devuelve una ConcreteFunction correspondiente a la firma de la función.
Gráfico
gráfico ()
Devuelve el gráfico que describe el cálculo realizado por el modelo.
Paquete de modelos guardados estático
cargar (etiquetas String exportDir, String...)
Cargue un modelo guardado desde un directorio de exportación.
estático SavedModelBundle.Loader
cargador (cadena 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

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

Invoca la función predeterminada directamente desde este modelo.

La selección de 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 se asigna al nombre de firma predeterminado, que es "serving_default".

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
Excepción de argumento ilegal si no se puede seleccionar ninguna función por defecto

cierre público vacío ()

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

Exportador público estático SavedModelBundle.Exporter (String exportDir)

Exportar un modelo guardado.

Devuelve un objeto Exporter para establecer 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 (Clave de firma de cadena)

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

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

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

gráfico gráfico público ()

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

Carga pública estática de SavedModelBundle (etiquetas String exportDir, String...)

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 el metagraphdef específico a cargar.
Devoluciones
  • un paquete que contiene el gráfico y la sesión asociada.

Cargador público estático 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.

metaGraphDef público 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.