Transformers en PyTorch
Ahora que ya conoces los distintos componentes de la arquitectura transformer, ¡toca definir uno! El módulo torch.nn, que hemos importado como nn, ofrece una forma muy cómoda de hacerlo en solo unas pocas líneas de código.
Este ejercicio forma parte del curso
Modelos Transformer con PyTorch
Instrucciones del ejercicio
- Define un transformer con
8cabezas de atención,6capas de codificador y decodificador, y para incrustaciones de secuencias de entrada de longitud1536. - Imprime el objeto del modelo para ver la arquitectura.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# Define the transformer model
model = ____
# Print the model object
print(____)