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
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)