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

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

Xem khóa học

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)
Chỉnh sửa và Chạy Mã