LoslegenKostenlos starten

PyTorch-Transformer

Jetzt, da du mit den verschiedenen Komponenten der Transformer-Architektur vertraut bist, wird es Zeit, einen zu definieren! Das Modul torch.nn, das für dich als nn importiert wurde, bietet eine wirklich elegante Möglichkeit, das in nur wenigen Zeilen Code umzusetzen.

Diese Übung ist Teil des Kurses

<Kurs>Transformer-Modelle mit PyTorch</Kurs>
Kurs ansehen

Übungsanweisungen

  • Definiere einen Transformer mit 8 Attention-Heads, 6 Encoder- und Decoder-Layern und Eingabesequenz-Embeddings der Länge 1536.
  • Gib das Modellobjekt aus, um die Modellarchitektur anzuzeigen.

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

# Define the transformer model
model = ____

# Print the model object
print(____)
Code bearbeiten und ausführen