Chargez les paramètres d’intégration de Momentum avec prise en charge du débogage.
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 | LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | Attributs facultatifs pour LoadTPUEmbeddingMomentumParametersGradAccumDebug |
Méthodes publiques
statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | config (configuration de chaîne) |
Charge statiqueTPUEmbeddingMomentumParametersGradAccumDebug | créer (portée de portée , paramètres d'opérande <Float>, instanta d'opérande <Float>, accumulateurs de gradient d'opérande <Float>, numShards longs, ID de fragment long, options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingMomentumParametersGradAccumDebug. |
statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | ID de table (ID de table longue) |
statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options | tableName (String tableName) |
Méthodes héritées
Méthodes publiques
configuration publique statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options (configuration de chaîne)
public static LoadTPUEmbeddingMomentumParametersGradAccumDebug créer (portée de portée , paramètres d'opérande <Float>, momenta d'opérande <Float>, opérandes <Float> gradientAccumulators, options Long numShards, Long shardId, Options... )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingMomentumParametersGradAccumDebug.
Paramètres
portée | portée actuelle |
---|---|
paramètres | Valeur des paramètres utilisés dans l'algorithme d'optimisation Momentum. |
moment | Valeur de l'impulsion utilisée dans l'algorithme d'optimisation Momentum. |
gradientAccumulateurs | Valeur de gradient_accumulators utilisée dans l'algorithme d'optimisation Momentum. |
choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de LoadTPUEmbeddingMomentumParametersGradAccumDebug