CommencerCommencer gratuitement

Transformers PyTorch

Maintenant que vous connaissez les différents composants de l’architecture Transformer, il est temps d’en définir un ! Le module torch.nn, importé pour vous sous le nom nn, offre une manière très pratique de le faire en seulement quelques lignes de code.

Cet exercice fait partie du cours

Modèles Transformer avec PyTorch

Afficher le cours

Instructions

  • Définissez un transformer avec 8 têtes d’attention, 6 couches d’encodeur et de décodeur, et des embeddings de séquences 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 cet exemple de code.

# Define the transformer model
model = ____

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