RetrieveTPUEmbeddingMomentumParameters

classe finale pubblica RetrieveTPUEmbeddingMomentumParameters

Recupera i parametri di incorporamento di Momentum.

Un'operazione che recupera i parametri di ottimizzazione dall'incorporamento nella memoria dell'host. Deve essere preceduto da un'operazione ConfigureTPUEmbeddingHost che imposta la corretta configurazione della tabella di incorporamento. Ad esempio, questa operazione viene utilizzata per recuperare parametri aggiornati prima di salvare un checkpoint.

Classi nidificate

classe RecuperaTPUEmbeddingMomentumParameters.Options Attributi facoltativi per RetrieveTPUEmbeddingMomentumParameters

Metodi pubblici

static RetrieveTPUEmbeddingMomentumParameters.Options
configurazione (configurazione stringa)
Recupero staticoTPUEmbeddingMomentumParameters
crea (ambito ambito , numShards lunghi, shardId lungo, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione RetrieveTPUEmbeddingMomentumParameters.
Uscita <Mobile>
momento ()
Parametro momento aggiornato dall'algoritmo di ottimizzazione Momentum.
Uscita <Mobile>
parametri ()
Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum.
static RetrieveTPUEmbeddingMomentumParameters.Options
tableId (Tabella lunga)
static RetrieveTPUEmbeddingMomentumParameters.Options
tableName (Stringa tableName)

Metodi ereditati

Metodi pubblici

Configurazione RetrieveTPUEmbeddingMomentumParameters.Options pubblica statica (configurazione stringa)

creazione statica pubblica RetrieveTPUEmbeddingMomentumParameters (ambito ambito , numShards lunghi, shardId lungo, Opzioni... opzioni)

Metodo factory per creare una classe che racchiude una nuova operazione RetrieveTPUEmbeddingMomentumParameters.

Parametri
scopo ambito attuale
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di RetrieveTPUEmbeddingMomentumParameters

uscita pubblica <Float> momento ()

Parametro momento aggiornato dall'algoritmo di ottimizzazione Momentum.

parametri <Float> di output pubblico ()

Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum.

public static RetrieveTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingMomentumParameters.Options tableName (String tableName)