LoadTPUEmbeddingCenteredRMSPropParameters

public final class LoadTPUEmbeddingCenteredRMSPropParameters

Parametri di incorporamento RMSProp centrati sul carico.

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

Metodi pubblici

static LoadTPUEmbeddingCenteredRMSPropParameters.Options
config (String config)
static LoadTPUEmbeddingCenteredRMSPropParameters
create ( ambito ambito, parametri operando <Float>, operando <Float> ms, operando <Float> mom, operando <Float> mg, long numShards, long shardId, opzioni ...
Metodo Factory per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingCenteredRMSPropParameters.
static LoadTPUEmbeddingCenteredRMSPropParameters.Options
tableId (Long tableId)
static LoadTPUEmbeddingCenteredRMSPropParameters.Options
tableName (stringa tableName)

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options config (String config)

public static LoadTPUEmbeddingCenteredRMSPropParameters create ( Scope scope, Operand <Float> parameters, Operand <Float> ms, Operand <Float> mom, Operand <Float> mg, Long numShards, Long shardId, Options ... options)

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

Parametri
scopo ambito attuale
parametri Valore dei parametri utilizzati nell'algoritmo di ottimizzazione RMSProp centrato.
SM Valore di ms utilizzato nell'algoritmo di ottimizzazione RMSProp centrato.
mamma Valore di mom utilizzato nell'algoritmo di ottimizzazione RMSProp centrato.
mg Valore di mg utilizzato nell'algoritmo di ottimizzazione RMSProp centrato.
opzioni trasporta valori di attributi opzionali
ritorna
  • una nuova istanza di LoadTPUEmbeddingCenteredRMSPropParameters

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingCenteredRMSPropParameters.Options tableName (String tableName)