Bắt đầu ngayBắt đầu miễn phí

Transformers trong PyTorch

Giờ bạn đã quen với các thành phần của kiến trúc transformer, hãy cùng định nghĩa một mô hình nhé! Mô-đun torch.nn, đã được nhập sẵn với tên nn, cung cấp một cách rất tiện để làm điều này chỉ với vài dòng mã.

Bài tập này là một phần của khóa học

Các mô hình Transformer với PyTorch

Xem khóa học

Hướng dẫn bài tập

  • Định nghĩa một transformer với 8 đầu attention, 6 lớp encoder và decoder, và embedding cho chuỗi đầu vào có độ dài 1536.
  • In đối tượng mô hình để xem kiến trúc mô hình.

Bài tập tương tác thực hành trực tiếp

Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.

# Define the transformer model
model = ____

# Print the model object
print(____)
Chỉnh sửa và Chạy Mã