MulaiMulai sekarang secara gratis

Transformer di PyTorch

Sekarang Anda telah memahami berbagai komponen arsitektur transformer, saatnya mendefinisikannya! Modul torch.nn, yang telah diimpor sebagai nn, menyediakan cara yang sangat praktis untuk melakukannya hanya dalam beberapa baris kode.

Latihan ini adalah bagian dari kursus

Model Transformer dengan PyTorch

Lihat Kursus

Petunjuk latihan

  • Definisikan sebuah transformer dengan 8 attention head, 6 lapisan encoder dan decoder, serta embedding urutan masukan dengan panjang 1536.
  • Cetak objek model untuk melihat arsitektur model.

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