SavedModelBundle

classe publique SavedModelBundle

SavedModelBundle représente un modèle chargé depuis le stockage.

Le modèle consiste en une description du calcul (un Graph ), une Session avec des tenseurs (par exemple, des paramètres ou des variables dans le graphe) initialisés aux valeurs enregistrées en stockage, et une description du modèle en tant que tampon de protocole MetaGraphDef .

Classes imbriquées

classe SavedModelBundle.Exporter Options d'exportation d'un SavedModel.
classe SavedModelBundle.Loader Options de chargement d'un SavedModel.

Constantes

Chaîne DEFAULT_TAG

Méthodes publiques

Carte<String, Tensor >
appel (Map<String, Tensor > arguments)
Appelle la fonction par défaut directement à partir de ce modèle.
vide
fermer ()
Libère les ressources (le Graph et Session ) associées au bundle de modèles enregistré.
statique SavedModelBundle.Exporter
exportateur (String exportDir)
Exportez un modèle enregistré.
Fonction Béton
fonction (String signatureKey)
Renvoie une ConcreteFunction correspondant à la signature de la fonction.
Graphique
graphique ()
Renvoie le graphique qui décrit le calcul effectué par le modèle.
statique SavedModelBundle
charger (balises String exportDir, String...)
Chargez un modèle enregistré à partir d'un répertoire d'exportation.
statique SavedModelBundle.Loader
chargeur (String exportDir)
Chargez un modèle enregistré.
MétaGraphDef
métaGraphDef ()
Renvoie le tampon de protocole MetaGraphDef associé au modèle enregistré.
Session
session ()
Renvoie la Session avec laquelle effectuer le calcul à l'aide du modèle.
Liste< Signature >
signature ()
Renvoie la signature de toutes les fonctions disponibles dans ce modèle enregistré.

Méthodes héritées

Constantes

Chaîne finale statique publique DEFAULT_TAG

Valeur constante : « servir »

Méthodes publiques

public Map<String, Tensor > appel (Map<String, Tensor > arguments)

Appelle la fonction par défaut directement à partir de ce modèle.

La sélection de fonction par défaut est effectuée en fonction de la première des conditions suivantes qui est vraie :

  • La fonction est la seule signature disponible attachée au graphique principal de ce modèle enregistré
  • La fonction est mappée sur le nom de signature par défaut, qui est "serving_default".

L'appelant est responsable de la fermeture de tous les Tensors renvoyés.

Paramètres
arguments liste des tenseurs d'entrée, mappés par leur nom de signature
Retours
  • liste des tenseurs de sortie, mappés par le nom de la signature
Jetés
IllegalArgumentException si aucune fonction ne peut être sélectionnée par défaut

public vide fermer ()

Libère les ressources (le Graph et Session ) associées au bundle de modèles enregistré.

exportateur public statique SavedModelBundle.Exporter (String exportDir)

Exportez un modèle enregistré.

Renvoie un objet Exporter pour définir les options de configuration avant d'enregistrer réellement le modèle.

Paramètres
exportDir le chemin du répertoire contenant un modèle enregistré.

fonction publique ConcreteFunction (String signatureKey)

Renvoie une ConcreteFunction correspondant à la signature de la fonction.

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

Paramètres
clé de signature nom du SignatureDef dans le modèle enregistré.
Retours
  • objet qui peut être utilisé pour appeler une fonction
Jetés
IllegalArgumentException si signatureKey n'est pas trouvé dans ce modèle enregistré.

graphique graphique public ()

Renvoie le graphique qui décrit le calcul effectué par le modèle.

Chargement statique public SavedModelBundle (balises String exportDir, String...)

Chargez un modèle enregistré à partir d'un répertoire d'exportation. Le modèle en cours de chargement doit être créé à l'aide de l' API du modèle enregistré .

Cette méthode est un raccourci pour :

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

Paramètres
exportDir le chemin du répertoire contenant un modèle enregistré.
balises les balises identifiant le métagraphdef spécifique à charger.
Retours
  • un bundle contenant le graphique et la session associée.

chargeur statique public SavedModelBundle.Loader (String exportDir)

Chargez un modèle enregistré.

Renvoie un objet Loader qui peut définir les options de configuration avant de charger réellement le modèle,

Paramètres
exportDir le chemin du répertoire contenant un modèle enregistré.

public MetaGraphDef métaGraphDef ()

Renvoie le tampon de protocole MetaGraphDef associé au modèle enregistré.

séance publique séance ()

Renvoie la Session avec laquelle effectuer le calcul à l'aide du modèle.

Retours
  • la session initialisée

Liste publique< Signature > signatures ()

Renvoie la signature de toutes les fonctions disponibles dans ce modèle enregistré.