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