CaricaTPUEmbeddingMomentumParametri

classe finale pubblica LoadTPUEmbeddingMomentumParameters

Carica i parametri di incorporamento di Momentum.

Un'operazione che carica i parametri di ottimizzazione in HBM per l'incorporamento. Deve essere preceduto da un'operazione ConfigureTPUEmbeddingHost che imposta la corretta configurazione della tabella di incorporamento. Ad esempio, questa operazione viene utilizzata per installare i parametri caricati da un checkpoint prima dell'esecuzione di un ciclo di training.

Classi nidificate

classe LoadTPUEmbeddingMomentumParameters.Options Attributi facoltativi per LoadTPUEmbeddingMomentumParameters

Metodi pubblici

LoadTPUEmbeddingMomentumParameters.Options statico
configurazione (configurazione stringa)
LoadTPUEmbeddingMomentumParameters statici
create (ambito dell'ambito , parametri dell'operando <Float>, momento dell'operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParameters.
LoadTPUEmbeddingMomentumParameters.Options statico
tableId (Tabella lunga)
LoadTPUEmbeddingMomentumParameters.Options statico
tableName (Stringa tableName)

Metodi ereditati

Metodi pubblici

Configurazione LoadTPUEmbeddingMomentumParameters.Options pubblica statica (configurazione stringa)

creazione di parametri public static LoadTPUEmbeddingMomentumParameters (ambito ambito , parametri operando <Float>, momenta operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)

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

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione Momentum.
momento Valore della quantità di moto utilizzato nell'algoritmo di ottimizzazione della quantità di moto.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingMomentumParameters

public static LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)