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
Petunjuk 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 praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Define the transformer model
model = ____
# Print the model object
print(____)