ComeçarComece de graça

Transformers no PyTorch

Agora que você conhece os diferentes componentes da arquitetura de transformers, é hora de definir um! O módulo torch.nn, importado para você como nn, oferece uma forma bem prática de fazer isso em apenas algumas linhas de código.

Este exercício faz parte do curso

Modelos Transformer com PyTorch

Ver curso

Instruções do exercício

  • Defina um transformer com 8 cabeças de atenção, 6 camadas de codificador e decodificador, e embeddings de sequência de entrada com comprimento 1536.
  • Imprima o objeto do modelo para visualizar a arquitetura do modelo.

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