RecuperaTPUEmbeddingADAMParametersGradAccumDebug

public final class RetrieveTPUEmbeddingADAMParametersGradAccumDebug

Recupera i parametri di incorporamento di ADAM con il 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 RecuperaTPUEmbeddingADAMParametersGradAccumDebug.Options Attributi facoltativi per RetrieveTPUEmbeddingADAMParametersGradAccumDebug

Metodi pubblici

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

Metodi ereditati

Metodi pubblici

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options config (String config)

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

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

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

public Output <Float> gradienteAccumulatori ()

Parametro gradiente_accumulatori aggiornato dall'algoritmo di ottimizzazione di ADAM.

output pubblico <Float> Momenta ()

Momento dei parametri aggiornato dall'algoritmo di ottimizzazione di ADAM.

parametri di output pubblici <Float> ()

Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione di ADAM.

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (String tableName)

public Output <Float> velocità ()

Velocità dei parametri aggiornate dall'algoritmo di ottimizzazione di ADAM.