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.