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 (una rappresentazione serializzata di un buffer del protocollo MetaGraphDef ).

Classi nidificate

classe SavedModelBundle.Loader Opzioni per caricare un SavedModel.

Metodi pubblici

vuoto
vicino ()
Rilascia le risorse (il Graph e Session ) associate al bundle di modelli salvato.
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.
byte[]
metaGraphDef ()
Restituisce il buffer del protocollo MetaGraphDef serializzato associato al modello salvato.
Sessione
sessione ()
Restituisce la Session con cui eseguire il calcolo utilizzando il modello.

Metodi ereditati

Metodi pubblici

pubblico vuoto chiudi ()

Rilascia le risorse (il Graph e Session ) associate al bundle di modelli 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.

byte pubblico[] metaGraphDef ()

Restituisce il buffer del protocollo MetaGraphDef serializzato associato al modello salvato.

sessione pubblica ()

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

Ritorni
  • la sessione inizializzata