SavedModelBundle

classe publique SavedModelBundle

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

Le modèle se compose d'une description du calcul (un Graph ), d'une Session avec des tenseurs (par exemple, des paramètres ou des variables dans le graphique) initialisés aux valeurs enregistrées en stockage, et d'une description du modèle (une représentation sérialisée d'un tampon de protocole MetaGraphDef ).

Classes imbriquées

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

Méthodes publiques

vide
fermer ()
Libère les ressources (le Graph et Session ) associées au bundle de modèles enregistré.
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é.
octet[]
métaGraphDef ()
Renvoie le tampon de protocole MetaGraphDef sérialisé associé au modèle enregistré.
Session
session ()
Renvoie la Session avec laquelle effectuer le calcul à l'aide du modèle.

Méthodes héritées

Méthodes publiques

public vide fermer ()

Libère les ressources (le Graph et Session ) associées au bundle de modèles 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é.

octet public[] métaGraphDef ()

Renvoie le tampon de protocole MetaGraphDef sérialisé 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