Partecipa al simposio Women in ML il 7 dicembre Registrati ora

SavedModelBundle

classe pubblica SavedModelBundle

SavedModelBundle rappresenta un modello caricato dalla memoria.

Il modello consiste in una descrizione del calcolo (un Graph ), una Session con tensori (ad es. Parametri o variabili nel grafico) inizializzata su valori salvati in memoria e una descrizione del modello (una rappresentazione serializzata di un buffer di protocollo MetaGraphDef ).

Classi annidate

classe SavedModelBundle.Loader Opzioni per il caricamento di un SavedModel.

Metodi pubblici

vuoto
chiudere ()
Rilascia le risorse (il Graph e la Session ) associate al bundle di modelli salvato.
Grafico
grafico ()
Restituisce il grafico che descrive il calcolo eseguito dal modello.
static SavedModelBundle
caricare (String exportDir, String ... tag)
Carica un modello salvato da una directory di esportazione.
static SavedModelBundle.Loader
loader (String 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

public void close ()

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

public Graph graph ()

Restituisce il grafico che descrive il calcolo eseguito dal modello.

public static carico di SavedModelBundle (stringa 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 è un'abbreviazione per:

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

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

public static SavedModelBundle.Loader loader (String exportDir)

Carica un modello salvato.

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

Parametri
exportDir il percorso della directory contenente un modello salvato.

public byte [] metaGraphDef ()

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

pubblica Session session ()

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

ritorna
  • la sessione inizializzata