Mulai sekarangMulai 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 merupakan bagian dari kursus

Model Transformer dengan PyTorch

Lihat Kursus

Instruksi 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 langsung praktik

Cobalah latihan ini dengan melengkapi kode contoh ini.

# Define the transformer model
model = ____

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