パブリック最終クラスLoadTPUEmbeddingFrequencyEstimatorParameters
周波数推定器の埋め込みパラメータを読み込みます。
埋め込みのために最適化パラメーターを HBM にロードする操作。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作は、トレーニング ループが実行される前にチェックポイントからロードされるパラメーターをインストールするために使用されます。
ネストされたクラス
クラス | LoadTPUEmbeddingFrequencyEstimatorParameters.Options | LoadTPUEmbeddingFrequencyEstimatorParameters のオプションの属性 |
パブリックメソッド
静的LoadTPUEmbeddingFrequencyEstimatorParameters.Options | config (文字列構成) |
静的LoadTPUEmbeddingFrequencyEstimatorParameters | create (スコープスコープ、オペランド<Float> パラメーター、オペランド<Float> lastHitStep、Long numShards、Long shardId、 Options...オプション) 新しい LoadTPUEmbeddingFrequencyEstimatorParameters オペレーションをラップするクラスを作成するためのファクトリ メソッド。 |
静的LoadTPUEmbeddingFrequencyEstimatorParameters.Options | tableId (長いテーブル ID) |
静的LoadTPUEmbeddingFrequencyEstimatorParameters.Options | テーブル名(文字列テーブル名) |
継承されたメソッド
パブリックメソッド
public static LoadTPUEmbeddingFrequencyEstimatorParameters create (スコープスコープ、オペランド<Float> パラメーター、オペランド<Float> lastHitStep、Long numShards、Long shardId、 Options...オプション)
新しい LoadTPUEmbeddingFrequencyEstimatorParameters オペレーションをラップするクラスを作成するためのファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
パラメーター | 周波数推定最適化アルゴリズムで使用されるパラメーターの値。 |
最後のヒットステップ | 周波数推定最適化アルゴリズムで使用される last_hit_step の値。 |
オプション | オプションの属性値を持ちます |
戻り値
- LoadTPUEmbeddingFrequencyEstimatorParameters の新しいインスタンス