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
Instruções do exercício
- 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.
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(____)