パブリック最終クラスRetrieveTPUEmbeddingADAMParameters
ADAM埋め込みパラメータを取得します。
ホストメモリへの埋め込みから最適化パラメータを取得する操作。正しい埋め込みテーブル構成をセットアップするConfigureTPUEmbeddingHostopを前に付ける必要があります。たとえば、この操作は、チェックポイントを保存する前に更新されたパラメータを取得するために使用されます。
ネストされたクラス
クラス | RetrieveTPUEmbeddingADAMParameters.Options | RetrieveTPUEmbeddingADAMParameters オプションの属性 |
パブリックメソッド
静的RetrieveTPUEmbeddingADAMParameters.Options | config (String config) |
静的RetrieveTPUEmbeddingADAMParameters | create (スコープスコープ、Long numShards、Long shardId、 Options ...オプション) 新しいRetrieveTPUEmbeddingADAMParameters操作をラップするクラスを作成するファクトリメソッド。 |
出力<フロート> | モーメント() ADAM最適化アルゴリズムによって更新されたパラメータモーメント。 |
出力<フロート> | パラメータ() ADAM最適化アルゴリズムによって更新されたパラメーターパラメーター。 |
静的RetrieveTPUEmbeddingADAMParameters.Options | tableId (長いtableId) |
静的RetrieveTPUEmbeddingADAMParameters.Options | tableName (String tableName) |
出力<フロート> | 速度() ADAM最適化アルゴリズムによって更新されたパラメーター速度。 |
継承されたメソッド
パブリックメソッド
public static RetrieveTPUEmbeddingADAMParameters create (スコープスコープ、Long numShards、Long shardId、 Options ...オプション)
新しいRetrieveTPUEmbeddingADAMParameters操作をラップするクラスを作成するファクトリメソッド。
パラメーター
範囲 | 現在のスコープ |
---|---|
オプション | オプションの属性値を運ぶ |
戻り値
- RetrieveTPUEmbeddingADAMParametersの新しいインスタンス