IniziaInizia gratis

Transformer in PyTorch

Ora che conosci i diversi componenti dell'architettura dei transformer, è il momento di definirne uno! Il modulo torch.nn, importato per te come nn, offre un modo davvero pratico per farlo in poche righe di codice.

Questo esercizio fa parte del corso

Modelli Transformer con PyTorch

Visualizza il corso

Istruzioni dell'esercizio

  • Definisci un transformer con 8 teste di attenzione, 6 livelli di encoder e decoder, e per embedding di sequenze in input di lunghezza 1536.
  • Stampa l'oggetto del modello per visualizzare l'architettura del modello.

Esercizio pratico interattivo

Prova a risolvere questo esercizio completando il codice di esempio.

# Define the transformer model
model = ____

# Print the model object
print(____)
Modifica ed esegui il codice