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

Constantes

Chaîne OP_NAME Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Méthodes publiques

statique LoadTPUEmbeddingMomentumParameters.Options
config (configuration de chaîne)
statique LoadTPUEmbeddingMomentumParameters
créer ( Scope scope, Operand < TFloat32 > paramètres, Operand < TFloat32 > momenta, Long numShards, Long shardId, Options... 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

Constantes

chaîne finale statique publique OP_NAME

Le nom de cette opération, tel que connu par le moteur principal TensorFlow

Valeur constante : "LoadTPUEmbeddingMomentumParameters"

Méthodes publiques

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

public static LoadTPUEmbeddingMomentumParameters créer ( Scope scope, Operand < TFloat32 > paramètres, Operand < TFloat32 > momenta, Long numShards, Long shardId, Options... 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)