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 interattivo pratico
Prova questo esercizio completando questo codice di esempio.
# Define the transformer model
model = ____
# Print the model object
print(____)