LoadTPUEmbeddingADAMParameters

classe finale publique LoadTPUEmbeddingADAMParameters

Chargez les paramètres d’intégration ADAM.

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

Constantes

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

Méthodes publiques

statique LoadTPUEmbeddingADAMParameters.Options
config (configuration de chaîne)
static LoadTPUEmbeddingADAMParameters
créer ( Portée de portée, Paramètres de l'Opérande < TFloat32 >, Momenta de l'Opérande < TFloat32 >, Vitesses de l'Opérande < TFloat32 >, Long numShards, Long shardId, Options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération LoadTPUEmbeddingADAMParameters.
statique LoadTPUEmbeddingADAMParameters.Options
ID de table (ID de table longue)
statique LoadTPUEmbeddingADAMParameters.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 : "LoadTPUEmbeddingADAMParameters"

Méthodes publiques

configuration publique statique LoadTPUEmbeddingADAMParameters.Options (configuration de chaîne)

public static LoadTPUEmbeddingADAMParameters créer ( portée de portée , paramètres de l'opérande < TFloat32 >, impulsion de l'opérande < TFloat32 >, vitesses de l'opérande < TFloat32 >, numShards longs, ID de fragment long, options... options)

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

Paramètres
portée portée actuelle
paramètres Valeur des paramètres utilisés dans l'algorithme d'optimisation ADAM.
moment Valeur de l'impulsion utilisée dans l'algorithme d'optimisation ADAM.
vitesses Valeur des vitesses utilisées dans l'algorithme d'optimisation ADAM.
choix porte des valeurs d'attributs facultatifs
Retour
  • une nouvelle instance de LoadTPUEmbeddingADAMParameters

public statique LoadTPUEmbeddingADAMParameters.Options tableId (Long tableId)

public static LoadTPUEmbeddingADAMParameters.Options tableName (String tableName)