Kom igångKom igång gratis

Transformers med PyTorch

Nu när du är bekant med de olika komponenterna i transformer-arkitekturen är det dags att definiera en! Modulen torch.nn, importerad som nn, erbjuder ett smidigt sätt att göra detta på bara några kodrader.

Den här övningen är en del av kursen

Transformermodeller med PyTorch

Visa kurs

Övningsinstruktioner

  • Definiera en transformer med 8 uppmärksamhetshuvuden, 6 encoder- och decoderlager samt inmatningssekvenser med inbäddningar av längd 1536.
  • Skriv ut modellobjektet för att se modellarkitekturen.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

# Define the transformer model
model = ____

# Print the model object
print(____)
Redigera och kör kod