ComenzarEmpieza gratis

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

Ver curso

Instrucciones del ejercicio

  • Define un transformer con 8 cabezas de atención, 6 capas de codificador y decodificador, y para incrustaciones de secuencias de entrada de longitud 1536.
  • 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(____)
Editar y ejecutar código