public final class LoadTPUEmbeddingStochasticGradientDescentParameters
Carica i parametri di incorporamento SGD.
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 | LoadTPUEmbeddingStochasticGradientDescentParameters.Options | Attributi facoltativi per LoadTPUEmbeddingStochasticGradientDescentParameters | |
Metodi pubblici
| static LoadTPUEmbeddingStochasticGradientDescentParameters.Options | config (String config) |
| static LoadTPUEmbeddingStochasticGradientDescentParameters | create ( ambito ambito, parametri operando <Float>, long numShards, long shardId, opzioni ... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingStochasticGradientDescentParameters. |
| static LoadTPUEmbeddingStochasticGradientDescentParameters.Options | tableId (Long tableId) |
| static LoadTPUEmbeddingStochasticGradientDescentParameters.Options | tableName (stringa tableName) |
Metodi ereditati
Metodi pubblici
public static LoadTPUEmbeddingStochasticGradientDescentParameters create ( Scope scope, Operand <Float> parameters, Long numShards, Long shardId, Options ...
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingStochasticGradientDescentParameters.
Parametri
| scopo | ambito attuale |
|---|---|
| parametri | Valore dei parametri utilizzati nell'algoritmo di ottimizzazione della discesa del gradiente stocastico. |
| opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di LoadTPUEmbeddingStochasticGradientDescentParameters