Options d'exportation d'un SavedModel.
Méthodes publiques
vide | exporter () Enregistrez le modèle dans le répertoire d'exportation. |
SavedModelBundle.Exporter | |
SavedModelBundle.Exporter | withTags (String... balises) Définit l'ensemble de balises qui identifient le graphique spécifique à enregistrer dans le modèle enregistré. |
Méthodes héritées
Méthodes publiques
exportation vide publique ()
Enregistrez le modèle dans le répertoire d'exportation.
Jetés
IOException | si le modèle enregistré ou l'état de la variable ne peut pas être écrit sur le disque |
---|
public SavedModelBundle.Exporter withFunction (fonction ConcreteFunction )
Enregistrez une fonction concrète de ce modèle.
La fonction concrète porte une signature (c'est-à-dire une liste de noms d'entrées et de sorties conviviaux dans un graphique) et une session valide vers un graphique à enregistrer dans le modèle.
Remarque : À terme, TensorFlow pour Java prendra en charge l'exportation d'objets fonctions comme le fait l'API Python, mais pour le moment, seuls les modèles centrés sur la session sont pris en charge (c'est-à-dire les modèles comportant un seul graphique principal et une ou plusieurs signatures). Ces modèles sont compatibles avec ceux exportés par TensorFlow 1.x ou par les estimateurs TensorFlow 2.x.
Par conséquent, toutes les fonctions exportées dans un modèle doivent partager la même session pour le moment, sinon une exception sera levée.
Paramètres
fonction | une fonction portant une signature et une session valide au graphe à sauvegarder |
---|
Retour
- cet objet
Jetés
Exception d'argument illégal | si une fonction du même nom a déjà été ajoutée au modèle |
---|---|
Exception d'opération non prise en charge | si cette fonction ne partage pas la même session avec les autres fonctions ajoutées à ce modèle |
public SavedModelBundle.Exporter withTags (String... tags)
Définit l'ensemble de balises qui identifient le graphique spécifique à enregistrer dans le modèle enregistré.
Notez qu'un seul graphique par modèle peut être enregistré pour le moment à l'aide de cette API.
Paramètres
Mots clés | les balises identifiant le MetaGraphDef spécifique à enregistrer. |
---|
Retour
- cet objet
Jetés
Exception d'argument illégal | si les balises ne sont pas valides |
---|