Carica i parametri di incorporamento di Momentum.
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 | LoadTPUEmbeddingMomentumParameters.Options | Attributi facoltativi per LoadTPUEmbeddingMomentumParameters |
Metodi pubblici
LoadTPUEmbeddingMomentumParameters.Options statico | configurazione (configurazione stringa) |
LoadTPUEmbeddingMomentumParameters statici | create (ambito dell'ambito , parametri dell'operando <Float>, momento dell'operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni) Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParameters. |
LoadTPUEmbeddingMomentumParameters.Options statico | tableId (Tabella lunga) |
LoadTPUEmbeddingMomentumParameters.Options statico | tableName (Stringa tableName) |
Metodi ereditati
Metodi pubblici
creazione di parametri public static LoadTPUEmbeddingMomentumParameters (ambito ambito , parametri operando <Float>, momenta operando <Float>, numShards lunghi, shardId lungo, Opzioni... opzioni)
Metodo factory per creare una classe che racchiude una nuova operazione LoadTPUEmbeddingMomentumParameters.
Parametri
scopo | ambito attuale |
---|---|
parametri | Valore dei parametri utilizzati nell'algoritmo di ottimizzazione Momentum. |
momento | Valore della quantità di moto utilizzato nell'algoritmo di ottimizzazione della quantità di moto. |
opzioni | trasporta valori di attributi opzionali |
ritorna
- una nuova istanza di LoadTPUEmbeddingMomentumParameters