LoadTPUEmbeddingMomentumParametersGradAccumDebug

classe finale pubblica LoadTPUEmbeddingMomentumParametersGradAccumDebug

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

Valore costante: "LoadTPUEmbeddingMomentumParametersGradAccumDebug"

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

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)