Aan de slagGa gratis aan de slag

PyTorch Transformers

Nu je bekend bent met de verschillende componenten van de transformer-architectuur, is het tijd om er een te definiëren! De module torch.nn, voor je geïmporteerd als nn, biedt een heel handige manier om dit in slechts een paar regels code te doen.

Deze oefening maakt deel uit van de cursus

Transformermodels met PyTorch

Cursus bekijken

Oefeninstructies

  • Definieer een transformer met 8 attention heads, 6 encoder- en decoderlagen, en voor invoersequentie-embeddings van lengte 1536.
  • Print het modelobject om de modelarchitectuur te bekijken.

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# Define the transformer model
model = ____

# Print the model object
print(____)
Code bewerken en uitvoeren