MulaiMulai sekarang secara gratis

Menambahkan transformer head

Saatnya merancang sebuah transformer head yang dapat digunakan untuk tugas klasifikasi seperti analisis sentimen atau kategorisasi. Anda akan mendefinisikan kelas ClassifierHead, membuat instance untuk body dan head, lalu meneruskan serangkaian ID token melalui keduanya untuk menguji apakah keduanya bekerja seperti yang diharapkan.

Catatan: karena model ini belum dilatih, hasilnya tidak akan bermakna. Namun, menguji bahwa kode dapat memproses masukan dan menghasilkan keluaran dalam bentuk yang Anda harapkan merupakan pengujian yang baik.

Latihan ini adalah bagian dari kursus

Model Transformer dengan PyTorch

Lihat Kursus

Latihan interaktif praktis

Cobalah latihan ini dengan menyelesaikan kode contoh berikut.

# 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)
Edit dan Jalankan Kode