LoadTPUEmbeddingADAMParametersGradAccumDebug

public final class LoadTPUEmbeddingADAMParametersGradAccumDebug

Carica i parametri di incorporamento di ADAM con il supporto del 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 che venga eseguito un ciclo di addestramento.

Classi annidate

classe LoadTPUEmbeddingADAMParametersGradAccumDebug.Options Attributi facoltativi per LoadTPUEmbeddingADAMParametersGradAccumDebug

Metodi pubblici

static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
config (String config)
static LoadTPUEmbeddingADAMParametersGradAccumDebug
create ( Scope scope, Operando <Float> parametri, Operando <Float> Momenta, Operando <Float> velocità, Operando <Float> Gradiente Accumulatori, Long numShards, Long shardId, Opzioni ...
Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingADAMParametersGradAccumDebug.
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId (Long tableId)
static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options config (String config)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug create ( Scope scope, Operand <Float> parameters, Operand <Float> Momenta, Operand <Float> velocities, Operand <Float> gradientAccumulators, Long numShards, Long shardId, Options ...

Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingADAMParametersGradAccumDebug.

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione di ADAM.
Momenta Valore di Momenta utilizzato nell'algoritmo di ottimizzazione di ADAM.
velocità Valore delle velocità utilizzate nell'algoritmo di ottimizzazione di ADAM.
gradiente Accumulatori Valore di gradiente_accumulatori utilizzati nell'algoritmo di ottimizzazione di ADAM.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingADAMParametersGradAccumDebug

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (String tableName)