LoadTPUEmbeddingMomentumParameters

classe finale publique LoadTPUEmbeddingMomentumParameters

Chargez les paramètres d’intégration de Momentum.

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 LoadTPUEmbeddingMomentumParameters.Options Attributs facultatifs pour LoadTPUEmbeddingMomentumParameters

Méthodes publiques

statique LoadTPUEmbeddingMomentumParameters.Options
config (configuration de chaîne)
statique LoadTPUEmbeddingMomentumParameters
créer (portée de portée , paramètres d'opérande <Float>, momenta d'opérande <Float>, options Long numShards, Long shardId, Options... )
Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingMomentumParameters.
statique LoadTPUEmbeddingMomentumParameters.Options
ID de table (ID de table longue)
statique LoadTPUEmbeddingMomentumParameters.Options
tableName (String tableName)

Méthodes héritées

Méthodes publiques

Configuration publique statique LoadTPUEmbeddingMomentumParameters.Options (configuration de chaîne)

public static LoadTPUEmbeddingMomentumParameters créer (portée de portée , paramètres d'opérande <Float>, momenta d'opérande <Float>, options Long numShards, Long shardId, Options... )

Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingMomentumParameters.

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.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de LoadTPUEmbeddingMomentumParameters

public statique LoadTPUEmbeddingMomentumParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingMomentumParameters.Options tableName (String tableName)