Paramètres d’intégration RMSProp centrés sur la charge.
Une opération qui charge les paramètres d'optimisation dans HBM pour l'intégration. Doit être précédé d'une opération ConfigureTPUEmbeddingHost qui configure la configuration correcte de la table d'intégration. Par exemple, cette opération est utilisée pour installer les paramètres chargés à partir d'un point de contrôle avant l'exécution d'une boucle de formation.
Classes imbriquées
classe | LoadTPUEmbeddingCenteredRMSPropParameters.Options | Attributs facultatifs pour LoadTPUEmbeddingCenteredRMSPropParameters |
Constantes
Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
statique LoadTPUEmbeddingCenteredRMSPropParameters.Options | config (configuration de chaîne) |
statique LoadTPUEmbeddingCenteredRMSPropParameters | |
statique LoadTPUEmbeddingCenteredRMSPropParameters.Options | ID de table (ID de table longue) |
statique LoadTPUEmbeddingCenteredRMSPropParameters.Options | tableName (String tableName) |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
Configuration publique statique LoadTPUEmbeddingCenteredRMSPropParameters.Options (configuration de chaîne)
public static LoadTPUEmbeddingCenteredRMSPropParameters créer ( Portée de portée, Opérande < TFloat32 > paramètres, Opérande < TFloat32 > ms, Opérande < TFloat32 > mom, Opérande < TFloat32 > mg, Long numShards, Long shardId, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingCenteredRMSPropParameters.
Paramètres
portée | portée actuelle |
---|---|
paramètres | Valeur des paramètres utilisés dans l'algorithme d'optimisation centré RMSProp. |
MS | Valeur de ms utilisée dans l'algorithme d'optimisation centré RMSProp. |
maman | Valeur de mom utilisée dans l'algorithme d'optimisation centré RMSProp. |
mg | Valeur de mg utilisée dans l'algorithme d'optimisation centré RMSProp. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de LoadTPUEmbeddingCenteredRMSPropParameters