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
Hướng dẫn bài tập
- Định nghĩa một transformer với
8đầu attention,6lớp encoder và decoder, và embedding cho chuỗi đầu vào có độ dài1536. - 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(____)