LoadTPUEmbeddingRMSPropParameters

public final class LoadTPUEmbeddingRMSPropParameters

Carica i parametri di incorporamento di RMSProp.

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

Metodi pubblici

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

Metodi ereditati

Metodi pubblici

public static LoadTPUEmbeddingRMSPropParameters.Options config (String config)

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

Metodo di fabbrica per creare una classe che avvolge una nuova operazione LoadTPUEmbeddingRMSPropParameters.

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

public static LoadTPUEmbeddingRMSPropParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingRMSPropParameters.Options tableName (String tableName)