パブリック最終クラスLoadTPUEmbeddingADAMParametersGradAccumDebug
デバッグ サポートを使用して ADAM 埋め込みパラメータをロードします。
埋め込みのために最適化パラメーターを HBM にロードする操作。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作は、トレーニング ループが実行される前にチェックポイントからロードされるパラメーターをインストールするために使用されます。
ネストされたクラス
| クラス | LoadTPUEmbeddingADAMParametersGradAccumDebug.Options | LoadTPUEmbeddingADAMParametersGradAccumDebugのオプション属性 | |
定数
| 弦 | OP_NAME | TensorFlow コア エンジンによって認識される、この演算の名前 |
パブリックメソッド
継承されたメソッド
定数
パブリック静的最終文字列OP_NAME
TensorFlow コア エンジンによって認識される、この演算の名前
定数値: "LoadTPUEmbeddingADAMParametersGradAccumDebug"
パブリックメソッド
public static LoadTPUEmbeddingADAMParametersGradAccumDebug create (スコープスコープ、オペランド< TFloat32 > パラメーター、オペランド< TFloat32 > モーメンタ、オペランド< TFloat32 > 速度、オペランド< TFloat32 > gradientAccumulators、Long numShards、Long shardId、 Options...オプション)
新しい LoadTPUEmbeddingADAMParametersGradAccumDebug 操作をラップするクラスを作成するためのファクトリ メソッド。
パラメーター
| 範囲 | 現在のスコープ |
|---|---|
| パラメーター | ADAM 最適化アルゴリズムで使用されるパラメーターの値。 |
| 勢い | ADAM 最適化アルゴリズムで使用される運動量の値。 |
| 速度 | ADAM 最適化アルゴリズムで使用される速度の値。 |
| 勾配アキュムレータ | ADAM 最適化アルゴリズムで使用される gradient_accumulators の値。 |
| オプション | オプションの属性値を持ちます |
戻り値
- LoadTPUEmbeddingADAMParametersGradAccumDebug の新しいインスタンス