LoadTPUEmbeddingMomentumParametersGradAccumDebug

classe finale pubblica LoadTPUEmbeddingMomentumParametersGradAccumDebug

Carica i parametri di incorporamento di Momentum con il supporto per il debug.

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 LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Attributi facoltativi per LoadTPUEmbeddingMomentumParametersGradAccumDebug

Metodi pubblici

static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
configurazione (configurazione stringa)
carico staticoTPUEmbeddingMomentumParametersGradAccumDebug
create ( ambito ambito , parametri operando <Float>, momenta operando <Float>, gradientAccumulatori operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (Tabella lunga)
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableName (Stringa tableName)

Metodi ereditati

Metodi pubblici

configurazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options pubblica (configurazione stringa)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create ( ambito ambito , parametri operando <Float>, momenta operando <Float>, operando <Float> gradientAccumulators, numShards lunghi, shardId lungo, Opzioni... opzioni)

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

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.
gradientAccumulatori Valore di gradient_accumulators utilizzato nell'algoritmo di ottimizzazione Momentum.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingMomentumParametersGradAccumDebug

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)