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