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
Instruksi latihan
- Definisikan sebuah transformer dengan
8attention head,6lapisan encoder dan decoder, serta embedding urutan masukan dengan panjang1536. - 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(____)