LoadTPUEmbeddingADAMParametersGradAccumDebug

パブリック最終クラスLoadTPUEmbeddingADAMParametersGradAccumDebug

デバッグ サポートを使用して ADAM 埋め込みパラメータをロードします。

埋め込みのために最適化パラメーターを HBM にロードする操作。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作は、トレーニング ループが実行される前にチェックポイントからロードされるパラメーターをインストールするために使用されます。

ネストされたクラス

クラスLoadTPUEmbeddingADAMParametersGradAccumDebug.Options LoadTPUEmbeddingADAMParametersGradAccumDebugのオプション属性

定数

OP_NAME TensorFlow コア エンジンによって認識される、この演算の名前

パブリックメソッド

静的LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
config (文字列構成)
静的LoadTPUEmbeddingADAMParametersGradAccumDebug
create (スコープスコープ、オペランド< TFloat32 > パラメーター、オペランド< TFloat32 > モーメンタ、オペランド< TFloat32 > 速度、オペランド< TFloat32 > gradientAccumulators、Long numShards、Long shardId、オプション...オプション)
新しい LoadTPUEmbeddingADAMParametersGradAccumDebug 操作をラップするクラスを作成するためのファクトリ メソッド。
静的LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
tableId (長いテーブル ID)
静的LoadTPUEmbeddingADAMParametersGradAccumDebug.Options
テーブル名(文字列テーブル名)

継承されたメソッド

定数

パブリック静的最終文字列OP_NAME

TensorFlow コア エンジンによって認識される、この演算の名前

定数値: "LoadTPUEmbeddingADAMParametersGradAccumDebug"

パブリックメソッド

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options config (文字列構成)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug create (スコープスコープ、オペランド< TFloat32 > パラメーター、オペランド< TFloat32 > モーメンタ、オペランド< TFloat32 > 速度、オペランド< TFloat32 > gradientAccumulators、Long numShards、Long shardId、 Options...オプション)

新しい LoadTPUEmbeddingADAMParametersGradAccumDebug 操作をラップするクラスを作成するためのファクトリ メソッド。

パラメーター
範囲現在のスコープ
パラメーターADAM 最適化アルゴリズムで使用されるパラメーターの値。
勢いADAM 最適化アルゴリズムで使用される運動量の値。
速度ADAM 最適化アルゴリズムで使用される速度の値。
勾配アキュムレータADAM 最適化アルゴリズムで使用される gradient_accumulators の値。
オプションオプションの属性値を持ちます
戻り値
  • LoadTPUEmbeddingADAMParametersGradAccumDebug の新しいインスタンス

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParametersGradAccumDebug.Options tableName (String tableName)