Inizia subitoInizia 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 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 interattivo pratico

Prova questo esercizio completando questo codice di esempio.

# Define the transformer model
model = ____

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