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
Övningsinstruktioner
- Definiera en transformer med
8uppmärksamhetshuvuden,6encoder- och decoderlager samt inmatningssekvenser med inbäddningar av längd1536. - 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(____)