Recupera i parametri di incorporamento di Momentum con il supporto per il debug.
Un'operazione che recupera i parametri di ottimizzazione dall'incorporamento nella memoria dell'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 parametri aggiornati prima di salvare un checkpoint.
Classi nidificate
| classe | RecuperaTPUEmbeddingMomentumParametersGradAccumDebug.Options | Attributi facoltativi per RetrieveTPUEmbeddingMomentumParametersGradAccumDebug | |
Metodi pubblici
| static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options | configurazione (configurazione stringa) |
| static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug | crea (ambito ambito , numShards lunghi, shardId lungo, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione RetrieveTPUEmbeddingMomentumParametersGradAccumDebug. |
| Uscita <Mobile> | gradientAccumulatori () Parametro gradient_accumulators aggiornato dall'algoritmo di ottimizzazione Momentum. |
| Uscita <Mobile> | momento () Parametro momento aggiornato dall'algoritmo di ottimizzazione Momentum. |
| Uscita <Mobile> | parametri () Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum. |
| static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableId (Tabella lunga) |
| static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableName (Stringa tableName) |
Metodi ereditati
Metodi pubblici
configurazione pubblica statica RetrieveTPUEmbeddingMomentumParametersGradAccumDebug.Options (configurazione stringa)
public static RetrieveTPUEmbeddingMomentumParametersGradAccumDebug create (ambito ambito , numShards lunghi, shardId lungo, Opzioni... opzioni)
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
output pubblico <Float> gradientAccumulators ()
Parametro gradient_accumulators aggiornato dall'algoritmo di ottimizzazione Momentum.
uscita pubblica <Float> momento ()
Parametro momento aggiornato dall'algoritmo di ottimizzazione Momentum.
Parametri <Float> di output pubblico ()
Parametri dei parametri aggiornati dall'algoritmo di ottimizzazione Momentum.