LoadTPUEmbeddingRMSPropParametersGradAccumDebug

public final class LoadTPUEmbeddingRMSPropParametersGradAccumDebug

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

Classi annidate

classe LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options Attributi facoltativi per LoadTPUEmbeddingRMSPropParametersGradAccumDebug

Metodi pubblici

static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options
config (String config)
static LoadTPUEmbeddingRMSPropParametersGradAccumDebug
create ( Scope scope, Operand <Float> parameters, Operand <Float> ms, Operand <Float> mom, Operand <Float> gradientAccumulator, Long numShards, Long shardId, Options ... options)
Metodo factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingRMSPropParametersGradAccumDebug.
static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options
tableId (Long tableId)
static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options config (String config)

public static LoadTPUEmbeddingRMSPropParametersGradAccumDebug create ( Scope scope, Operand <Float> parameters, Operand <Float> ms, Operand <Float> mom, Operand <Float> gradientAccumulator, Long numShards, Long shardId, Options ...

Metodo factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingRMSPropParametersGradAccumDebug.

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione RMSProp.
SM Valore di ms utilizzato nell'algoritmo di ottimizzazione RMSProp.
mamma Valore di mom utilizzato nell'algoritmo di ottimizzazione RMSProp.
gradiente Accumulatori Valore degli accumulatori_ gradiente utilizzati nell'algoritmo di ottimizzazione RMSProp.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingRMSPropParametersGradAccumDebug

public static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingRMSPropParametersGradAccumDebug.Options tableName (String tableName)