Aan de slagGa gratis aan de slag

De transformer-head toevoegen

Tijd om een transformer-head te ontwerpen die je kunt gebruiken voor classificatietaken zoals sentimentanalyse of categorisatie. Je definieert een ClassifierHead-klasse, maakt instanties van de body en de head, en stuurt een reeks token-ID's door beide heen om te testen of ze werken zoals verwacht.

Let op: omdat dit model nog niet is getraind, zijn de outputwaarden zinloos. Het is echter een goede test om te controleren of de code invoer kan verwerken en output kan genereren in het verwachte formaat.

Deze oefening maakt deel uit van de cursus

Transformermodels met PyTorch

Cursus bekijken

Praktische interactieve oefening

Probeer deze oefening eens door deze voorbeeldcode in te vullen.

# 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)
Code bewerken en uitvoeren