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
Istruzioni dell'esercizio
- Definisci un transformer con
8teste di attenzione,6livelli di encoder e decoder, e per embedding di sequenze in input di lunghezza1536. - 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(____)