LoadTPUEmbeddingADAMParameters

public final class LoadTPUEmbeddingADAMParameters

Carica i parametri di incorporamento di ADAM.

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 LoadTPUEmbeddingADAMParameters.Options Attributi facoltativi per LoadTPUEmbeddingADAMParameters

Metodi pubblici

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

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingADAMParameters.Options config (String config)

public static LoadTPUEmbeddingADAMParameters create ( Scope scope, Operand <Float> parameters, Operand <Float> Momenta, Operand <Float> velocities, Long numShards, Long shardId, Options ... options)

Metodo factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingADAMParameters.

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.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingADAMParameters

public static LoadTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)