Aan de slagBegin gratis

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

Bekijk cursus

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.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

# Define the transformer model
model = ____

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