Carica i parametri di incorporamento di Momentum 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 dell'esecuzione di un ciclo di training.
Classi nidificate
classe | LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | Attributi facoltativi per LoadTPUEmbeddingMomentumParametersGradAccumDebug |
Costanti
Corda | OP_NAME | Il nome di questa operazione, come noto al motore principale di TensorFlow |
Metodi pubblici
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | configurazione (configurazione stringa) |
carico staticoTPUEmbeddingMomentumParametersGradAccumDebug | crea (ambito ambito , parametri Operando < TFloat32 >, Operando < TFloat32 > momenta, Operando < TFloat32 > gradientAccumulators, Long numShards, Long shardId, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParametersGradAccumDebug. |
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableId (Tabella lunga) |
static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableName (Stringa tableName) |
Metodi ereditati
Costanti
Stringa finale statica pubblica OP_NAME
Il nome di questa operazione, come noto al motore principale di TensorFlow
Metodi pubblici
configurazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options pubblica (configurazione stringa)
public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create ( ambito ambito , parametri Operando < TFloat32 >, Operando < TFloat32 > momenta, Operando < TFloat32 > gradientAccumulators, Long numShards, Long shardId, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParametersGradAccumDebug.
Parametri
scopo | ambito attuale |
---|---|
parametri | Valore dei parametri utilizzati nell'algoritmo di ottimizzazione Momentum. |
momento | Valore della quantità di moto utilizzato nell'algoritmo di ottimizzazione della quantità di moto. |
gradientAccumulatori | Valore di gradient_accumulators utilizzato nell'algoritmo di ottimizzazione Momentum. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di LoadTPUEmbeddingMomentumParametersGradAccumDebug