IniziaInizia gratis

PyTorch Transformers

Now you're familiar with the different components of the transformer architecture, it's time to define one! The torch.nn module, imported for you as nn, provides a really nice way to do this in just a few lines of code.

Questo esercizio fa parte del corso

Transformer Models with PyTorch

Visualizza il corso

Istruzioni dell'esercizio

  • Define a transformer with 8 attention heads, 6 encoder and decoder layers, and for input sequence embeddings of length 1536.
  • Print the model object to view the model architecture.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Define the transformer model
model = ____

# Print the model object
print(____)
Modifica ed esegui il codice