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 | |
static LoadTPUEmbeddingCenteredRMSPropParameters.Options | tableId (Long tableId) |
static LoadTPUEmbeddingCenteredRMSPropParameters.Options | tableName (stringa tableName) |
Metodi ereditati
Metodi pubblici
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