LoadTPUEmbeddingStochasticGradientDescentParameters

classe finale pubblica 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 dell'esecuzione di un ciclo di training.

Classi nidificate

classe LoadTPUEmbeddingStochasticGradientDescentParameters.Options Attributi facoltativi per LoadTPUEmbeddingStochasticGradientDescentParameters

Metodi pubblici

static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
configurazione (configurazione stringa)
parametri di caricamento staticoTPUEmbeddingStochasticGradientDescent
create (ambito ambito , parametri operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingStochasticGradientDescentParameters.
static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
tableId (Tabella lunga)
static LoadTPUEmbeddingStochasticGradientDescentParameters.Options
tableName (Stringa tableName)

Metodi ereditati

Metodi pubblici

configurazione LoadTPUEmbeddingStochasticGradientDescentParameters.Options pubblica (configurazione stringa)

creazione di parametri statici pubblici LoadTPUEmbeddingStochasticGradientDescentParameters (ambito ambito , parametri operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)

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

public static LoadTPUEmbeddingStochasticGradientDescentParameters.Options tableName (String tableName)