SavedModelBundle

classe pubblica SavedModelBundle

SavedModelBundle rappresenta un modello caricato dallo spazio di archiviazione.

Il modello consiste in una descrizione del calcolo (un Graph ), una Session con tensori (ad esempio, parametri o variabili nel grafico) inizializzati su valori salvati in memoria e una descrizione del modello come buffer del protocollo MetaGraphDef .

Classi nidificate

classe SavedModelBundle.Exporter Opzioni per esportare un modello salvato.
classe SavedModelBundle.Loader Opzioni per caricare un SavedModel.

Costanti

Corda DEFAULT_TAG

Metodi pubblici

Mappa<Stringa, Tensore >
chiamata (argomenti Map<String, Tensor >)
Richiama la funzione predefinita direttamente da questo modello.
vuoto
vicino ()
Rilascia le risorse (il Graph e Session ) associate al bundle di modelli salvato.
SavedModelBundle.Exporter statico
esportatore (Stringa exportDir)
Esporta un modello salvato.
FunzioneConcreta
funzione (Stringa SignatureKey)
Restituisce una ConcreteFunction corrispondente alla firma della funzione.
Grafico
grafico ()
Restituisce il grafico che descrive il calcolo eseguito dal modello.
SavedModelBundle statico
caricamento (stringa exportDir, stringa... tag)
Carica un modello salvato da una directory di esportazione.
SavedModelBundle.Loader statico
caricatore (Stringa exportDir)
Carica un modello salvato.
MetaGraphDef
metaGraphDef ()
Restituisce il buffer del protocollo MetaGraphDef associato al modello salvato.
Sessione
sessione ()
Restituisce la Session con cui eseguire il calcolo utilizzando il modello.
Elenco <Firma>
firme ()
Restituisce la firma di tutte le funzioni disponibili in questo modello salvato.

Metodi ereditati

Costanti

Stringa finale statica pubblica DEFAULT_TAG

Valore costante: "servire"

Metodi pubblici

chiamata pubblica Map<String, Tensor > (argomenti Map<String, Tensor >)

Richiama la funzione predefinita direttamente da questo modello.

La selezione della funzione predefinita viene effettuata in base alla prima delle seguenti condizioni che è vera:

  • La funzione è l'unica firma disponibile allegata al grafico principale di questo modello salvato
  • La funzione è mappata sul nome della firma predefinita, che è "serving_default"

Il chiamante è responsabile della chiusura di tutti i Tensor restituiti.

Parametri
argomenti elenco di tensori di input, mappati in base al nome della firma
Ritorni
  • elenco dei tensori di output, mappati in base al nome della firma
Lancia
IllegalArgumentException se nessuna funzione può essere selezionata per impostazione predefinita

pubblico vuoto chiudi ()

Rilascia le risorse (il Graph e Session ) associate al bundle di modelli salvato.

esportatore pubblico statico SavedModelBundle.Exporter (String exportDir)

Esporta un modello salvato.

Restituisce un oggetto Exporter per impostare le opzioni di configurazione prima di salvare effettivamente il modello.

Parametri
diresportazione il percorso della directory contenente un modello salvato.

funzione pubblica ConcreteFunction (String SignatureKey)

Restituisce una ConcreteFunction corrispondente alla firma della funzione.

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

Parametri
firmaKey nome della SignatureDef nel modello salvato.
Ritorni
  • oggetto che può essere utilizzato per effettuare chiamate a una funzione
Lancia
IllegalArgumentException se signatureKey non viene trovata in questo modello salvato.

grafico grafico pubblico ()

Restituisce il grafico che descrive il calcolo eseguito dal modello.

caricamento pubblico statico di SavedModelBundle (tag String exportDir, String...)

Carica un modello salvato da una directory di esportazione. Il modello che viene caricato deve essere creato utilizzando l' API del modello salvato .

Questo metodo è una scorciatoia per:

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

Parametri
diresportazione il percorso della directory contenente un modello salvato.
tag i tag che identificano la metagraphdef specifica da caricare.
Ritorni
  • un bundle contenente il grafico e la sessione associata.

caricatore pubblico statico SavedModelBundle.Loader (stringa exportDir)

Carica un modello salvato.

Restituisce un oggetto Loader che può impostare le opzioni di configurazione prima di caricare effettivamente il modello,

Parametri
diresportazione il percorso della directory contenente un modello salvato.

metaGraphDef pubblico metaGraphDef ()

Restituisce il buffer del protocollo MetaGraphDef associato al modello salvato.

sessione pubblica ()

Restituisce la Session con cui eseguire il calcolo utilizzando il modello.

Ritorni
  • la sessione inizializzata

Elenco pubblico< Firma > firme ()

Restituisce la firma di tutte le funzioni disponibili in questo modello salvato.