LoadTPUEmbeddingMomentumParametersGradAccumDebug

public final class LoadTPUEmbeddingMomentumParametersGradAccumDebug

Carica i parametri di incorporamento di Momentum con 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 che venga eseguito un ciclo di addestramento.

Classi annidate

classe LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options Attributi facoltativi per LoadTPUEmbeddingMomentumParametersGradAccumDebug

Metodi pubblici

static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
config (String config)
static LoadTPUEmbeddingMomentumParametersGradAccumDebug
create ( Scope scope, Operando <Float> parametri, Operando <Float> Momenta, Operando <Float> GradienteAccumulatori, Long numShards, Long shardId, Opzioni ... opzioni)
Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (Long tableId)
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options config (String config)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create ( Scope scope, Operand <Float> parameters, Operand <Float> Momenta, Operand <Float> gradientAccumulator, Long numShards, Long shardId, Options ... options)

Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione Momentum.
Momenta Valore di Momenta utilizzato nell'algoritmo di ottimizzazione Momentum.
gradiente Accumulatori Valore di gradiente_accumulatori utilizzati 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)