RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

public final class RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

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

Un'operazione che recupera i parametri di ottimizzazione dall'incorporamento alla memoria 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 i parametri aggiornati prima di salvare un checkpoint.

Classi annidate

classe RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options Attributi facoltativi per RetrieveTPUEmbeddingMomentumParametersGradAccumDebug

Metodi pubblici

static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
config (String config)
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug
create ( Scope scope, Long numShards, Long shardId, Options ... options)
Metodo factory per creare una classe che racchiude una nuova operazione RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.
Uscita <Float>
gradienteAccumulatori ()
Parametro gradiente_accumulatori aggiornato dall'algoritmo di ottimizzazione Momentum.
Uscita <Float>
momento ()
Parametro Momenta aggiornato dall'algoritmo di ottimizzazione Momentum.
Uscita <Float>
parametri ()
Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum.
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableId (Long tableId)
static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug create ( Scope scope, Long numShards, Long shardId, Options ... options)

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

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

public Output <Float> gradienteAccumulatori ()

Parametro gradiente_accumulatori aggiornato dall'algoritmo di ottimizzazione Momentum.

output pubblico <Float> Momenta ()

Parametro Momenta aggiornato dall'algoritmo di ottimizzazione Momentum.

parametri di output pubblici <Float> ()

Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum.

public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)