ComeçarComece de graça

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.

Este exercício faz parte do curso

Transformer Models with PyTorch

Ver curso

Instruções do exercício

  • 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.

Exercício interativo prático

Experimente este exercício completando este código de exemplo.

# Define the transformer model
model = ____

# Print the model object
print(____)
Editar e executar o código