LoadTPUEmbeddingMomentumParametersGradAccumDebug

publiczna klasa końcowa LoadTPUEmbeddingMomentumParametersGradAccumDebug

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
utwórz (zakres zakresu , parametry argumentu <Float>, moment argumentu <Float>, argument <Float> gradientAccumulators, długie numShards, długie shardId, opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację LoadTPUEmbeddingMomentumParametersGradAccumDebug.
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.Options config (konfiguracja ciągu)

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

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Długi identyfikator stołu)

public static LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)