パブリック最終クラスLoadTPUEmbeddingFTRLParametersGradAccumDebug
デバッグサポートを使用して FTRL 埋め込みパラメータをロードします。
埋め込みのために最適化パラメーターを HBM にロードする操作。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作は、トレーニング ループが実行される前にチェックポイントからロードされるパラメーターをインストールするために使用されます。
ネストされたクラス
クラス | LoadTPUEmbeddingFTRLParametersGradAccumDebug.Options | LoadTPUEmbeddingFTRLParametersGradAccumDebug のオプション属性 |
パブリックメソッド
継承されたメソッド
パブリックメソッド
public static LoadTPUEmbeddingFTRLParametersGradAccumDebug create (スコープスコープ、オペランド<Float>パラメータ、オペランド<Float>アキュムレータ、オペランド<Float>線形、オペランド<Float>gradientAccumulators、Long numShards、Long shardId、 Options...オプション)
新しい LoadTPUEmbeddingFTRLParametersGradAccumDebug オペレーションをラップするクラスを作成するファクトリ メソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
パラメーター | FTRL 最適化アルゴリズムで使用されるパラメーターの値。 |
アキュムレータ | FTRL 最適化アルゴリズムで使用されるアキュムレータの値。 |
リニア | FTRL 最適化アルゴリズムで使用される線形の値。 |
勾配アキュムレータ | FTRL 最適化アルゴリズムで使用される gradient_accumulators の値。 |
オプション | オプションの属性値を持ちます |
戻り値
- LoadTPUEmbeddingFTRLParametersGradAccumDebug の新しいインスタンス