LoadTPUEmbeddingMomentumParameters

パブリック最終クラスLoadTPUEmbeddingMomentumParameters

Momentum 埋め込みパラメータをロードします。

埋め込みのために最適化パラメーターを HBM にロードする操作。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作は、トレーニング ループが実行される前にチェックポイントからロードされるパラメーターをインストールするために使用されます。

ネストされたクラス

クラスLoadTPUEmbeddingMomentumParameters.Options LoadTPUEmbeddingMomentumParametersのオプションの属性

パブリックメソッド

静的LoadTPUEmbeddingMomentumParameters.Options
config (文字列構成)
静的LoadTPUEmbeddingMomentumParameters
create (スコープスコープ、オペランド<Float> パラメータ、オペランド<Float> モーメンタ、Long numShards、Long shardId、 Options...オプション)
新しい LoadTPUEmbeddingMomentumParameters 操作をラップするクラスを作成するためのファクトリ メソッド。
静的LoadTPUEmbeddingMomentumParameters.Options
tableId (長いテーブル ID)
静的LoadTPUEmbeddingMomentumParameters.Options
テーブル名(文字列テーブル名)

継承されたメソッド

パブリックメソッド

public static LoadTPUEmbeddingMomentumParameters.Options構成(文字列構成)

public static LoadTPUEmbeddingMomentumParameters create (スコープスコープ、オペランド<Float> パラメータ、オペランド<Float> モーメンタ、Long numShards、Long shardId、 Options...オプション)

新しい LoadTPUEmbeddingMomentumParameters 操作をラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
パラメーター運動量最適化アルゴリズムで使用されるパラメータの値。
勢い運動量最適化アルゴリズムで使用される運動量の値。
オプションオプションの属性値を持ちます
戻り値
  • LoadTPUEmbeddingMomentumParameters の新しいインスタンス

public static LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)