LoadTPUEmbeddingMomentumParameters

public final class LoadTPUEmbeddingMomentumParameters

Carica i parametri di incorporamento di Momentum.

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

Metodi pubblici

static LoadTPUEmbeddingMomentumParameters.Options
config (String config)
static LoadTPUEmbeddingMomentumParameters
create ( Scope scope, Operando <Float> parameters, Operando <Float> Momenta, Long numShards, Long shardId, Options ...
Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingMomentumParameters.
static LoadTPUEmbeddingMomentumParameters.Options
tableId (Long tableId)
static LoadTPUEmbeddingMomentumParameters.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingMomentumParameters.Options config (String config)

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

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

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione Momentum.
Momenta Valore di Momenta utilizzato nell'algoritmo di ottimizzazione Momentum.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingMomentumParameters

public static LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)