MulaiMulai sekarang secara gratis

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.

Latihan ini adalah bagian dari kursus

Transformer Models with PyTorch

Lihat Kursus

Petunjuk latihan

  • Define a transformer with 8 attention heads, 6 encoder and decoder layers, and for input sequence embeddings of length 1536.
  • Print the model object to view the model architecture.

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# Define the transformer model
model = ____

# Print the model object
print(____)
Edit dan Jalankan Kode