Thêm phần đầu (head) cho transformer
Đến lúc thiết kế một transformer head có thể dùng cho các bài toán phân loại như phân tích cảm xúc hoặc phân loại chủ đề. Bạn sẽ định nghĩa lớp ClassifierHead, tạo các instance cho phần thân (body) và phần đầu (head), rồi truyền một dãy các token ID qua cả hai để kiểm tra chúng hoạt động đúng như kỳ vọng.
Lưu ý: vì mô hình này chưa được huấn luyện, đầu ra sẽ chưa có ý nghĩa. Tuy vậy, việc kiểm tra xem mã có xử lý đầu vào và tạo ra đầu ra đúng định dạng bạn mong đợi hay không là một bài kiểm thử tốt.
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
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.
# Complete the classification head
class ClassifierHead(nn.Module):
def __init__(self, d_model, num_classes):
super().__init__()
self.fc = ____
def forward(self, x):
logits = self.fc(x)
return F.____(logits, dim=-1)