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