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 |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
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
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Metody publiczne
public static LoadTPUEmbeddingMomentumParametersGradAccumDebug create ( Zakres zakresu, Operand < TFloat32 > parametry, Operand < TFloat32 > momenta, Operand < TFloat32 > 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