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 > | |
vacío | |
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 | |
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
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 ()
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. |
---|
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.