LoadTPUEmbeddingMomentumParametersGradAccumDebug

classe finale publique LoadTPUEmbeddingMomentumParametersGradAccumDebug

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

public statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableId (Long tableId)

public statique LoadTPUEmbeddingMomentumParametersGradAccumDebug.Options tableName (String tableName)