ÎncepețiÎncepe gratuit

Transformere cu PyTorch

Acum că ești familiarizat cu componentele arhitecturii transformer, e timpul să definești unul! Modulul torch.nn, importat pentru tine ca nn, oferă o modalitate foarte elegantă de a face asta în doar câteva linii de cod.

Acest exercițiu face parte din cursul

Modele Transformer cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

  • Definește un transformer cu 8 capete de atenție, 6 straturi encoder și decoder, și încorporări ale secvenței de intrare de lungime 1536.
  • Afișează obiectul model pentru a vizualiza arhitectura acestuia.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Define the transformer model
model = ____

# Print the model object
print(____)
Editează și rulează codul