CommencezCommencez gratuitement

Transformers PyTorch

Maintenant que vous connaissez les différentes composantes de l'architecture Transformer, il est temps d'en définir une ! Le module torch.nn, importé pour vous sous le nom nn, offre une façon très simple de le faire en seulement quelques lignes de code.

Cette activité fait partie du cours

Modèles Transformer avec PyTorch

Voir le cours

Instructions de l’exercice

  • Définissez un transformer avec 8 têtes d'attention, 6 couches d'encodeur et de décodeur, et des plongements de séquence d'entrée de longueur 1536.
  • Affichez l'objet du modèle pour voir l'architecture du modèle.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Define the transformer model
model = ____

# Print the model object
print(____)
Modifier et exécuter le code