TPUEmbeddingADAMパラメータの取得

パブリック最終クラスRetrieveTPUEmbeddingADAMParameters

ADAM 埋め込みパラメータを取得します。

ホストメモリへの埋め込みから最適化パラメータを取得する演算。正しい埋め込みテーブル構成を設定する ConfigureTPUEmbeddingHost オペレーションを前に置く必要があります。たとえば、この操作はチェックポイントを保存する前に更新されたパラメータを取得するために使用されます。

ネストされたクラス

クラスTPUEmbeddingADAMParameters.Options の取得RetrieveTPUEmbeddingADAMParametersのオプションの属性

パブリックメソッド

静的RetrieveTPUEmbeddingADAMParameters.Options
config (文字列構成)
静的RetrieveTPUEmbeddingADAMParameters
create (スコープscope、Long numShards、Long shardId、オプション...オプション)
新しい RetrieveTPUEmbeddingADAMParameters 操作をラップするクラスを作成するためのファクトリ メソッド。
出力<浮動小数点数>
勢い()
ADAM 最適化アルゴリズムによって更新されるパラメーターの運動量。
出力<浮動小数点数>
パラメーター()
パラメータ ADAM 最適化アルゴリズムによって更新されるパラメータ。
静的RetrieveTPUEmbeddingADAMParameters.Options
tableId (長いテーブル ID)
静的RetrieveTPUEmbeddingADAMParameters.Options
テーブル名(文字列テーブル名)
出力<浮動小数点数>
速度()
ADAM 最適化アルゴリズムによって更新されたパラメーター速度。

継承されたメソッド

パブリックメソッド

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

public static RetrieveTPUEmbeddingADAMParameters create (スコープscope、Long numShards、Long shardId、 Options...オプション)

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

パラメーター
範囲現在のスコープ
オプションオプションの属性値を持ちます
戻り値
  • RetrieveTPUEmbeddingADAMParameters の新しいインスタンス

public出力<Float> momenta ()

ADAM 最適化アルゴリズムによって更新されるパラメーターの運動量。

public出力<Float>パラメータ()

パラメータ ADAM 最適化アルゴリズムによって更新されるパラメータ。

public static RetrieveTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static RetrieveTPUEmbeddingADAMParameters.Options tableName (String tableName)

public出力<Float>速度()

ADAM 最適化アルゴリズムによって更新されたパラメーター速度。