Comece agoraComece grátis

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 exercicio faz parte do curso

Modelos Transformer com PyTorch

Ver curso

Instruções do exercicio

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

exercicio interativo prático

Tente este exercicio completando este código de exemplo.

# Define the transformer model
model = ____

# Print the model object
print(____)
Editar e Executar Código