パブリック最終クラス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 create (スコープスコープ、オペランド<Float> パラメータ、オペランド<Float> モーメンタ、Long numShards、Long shardId、 Options...オプション)
新しい LoadTPUEmbeddingMomentumParameters 操作をラップするクラスを作成するためのファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| パラメーター | 運動量最適化アルゴリズムで使用されるパラメータの値。 |
| 勢い | 運動量最適化アルゴリズムで使用される運動量の値。 |
| オプション | オプションの属性値を持ちます |
戻り値
- LoadTPUEmbeddingMomentumParameters の新しいインスタンス