PyTorch Transformers
Now you're familiar with the different components of the transformer architecture, it's time to define one! The torch.nn module, imported for you as nn, provides a really nice way to do this in just a few lines of code.
Diese Übung ist Teil des Kurses
Transformer Models with PyTorch
Anleitung zur Übung
- Define a transformer with
8attention heads,6encoder and decoder layers, and for input sequence embeddings of length1536. - Print the model object to view the model architecture.
Interaktive Übung
Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.
# Define the transformer model
model = ____
# Print the model object
print(____)