Załaduj parametry osadzania Momentum z obsługą debugowania.
Operacja ładująca parametry optymalizacyjne do HBM w celu osadzenia. Musi być poprzedzony operacją ConfigureTPUEmbeddingHost, która konfiguruje poprawną konfigurację tabeli osadzania. Na przykład ta operacja służy do instalowania parametrów ładowanych z punktu kontrolnego przed wykonaniem pętli szkoleniowej.
Klasy zagnieżdżone
klasa | LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | Opcjonalne atrybuty dla LoadTPUEmbeddingMomentumParametersGradAccumDebug |
Metody publiczne
statyczne LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | konfiguracja (konfiguracja ciągu) |
obciążenie statyczneTPUEmbeddingMomentumParametersGradAccumDebug | |
statyczne LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableId (długi identyfikator stołu) |
statyczne LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | nazwa_tabeli (ciąg znaków nazwa_tabeli) |
Metody dziedziczone
Metody publiczne
public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create (zakres zakresu , parametry argumentu <Float>, moment argumentu argumentu <Float>, argument argumentu <Float> gradientAccumulators, Long numShards, Long shardId, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację LoadTPUEmbeddingMomentumParametersGradAccumDebug.
Parametry
zakres | aktualny zakres |
---|---|
parametry | Wartość parametrów wykorzystywanych w algorytmie optymalizacji Momentum. |
chwila | Wartość pędu wykorzystywana w algorytmie optymalizacji Momentum. |
akumulatory gradientowe | Wartość gradient_akumulatorów wykorzystana w algorytmie optymalizacji Momentum. |
opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja LoadTPUEmbeddingMomentumParametersGradAccumDebug