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
Instruções do exercicio
- Defina um transformer com
8cabeças de atenção,6camadas de codificador e decodificador, e embeddings de sequência de entrada com comprimento1536. - 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(____)