Stel de Trainer in

Je model vervangt complexe zinnen door eenvoudigere, toegankelijkere parafrases om de leesbaarheid van vertalingen te verbeteren. Zo kan het bijvoorbeeld de oorspronkelijke zin "The inclement weather conditions precipitated the postponement of the outdoor event" vereenvoudigen tot "The bad weather caused the outdoor event to be delayed." Bouw de Trainer om de training van je vertaaldienst voor natuurlijke taal voor te bereiden! Deze oefening duurt even om uit te voeren door de aanroep trainer.train().

Er is alvast wat data ingeladen:

model is een Transformer-model
dataset bevat de MRPC-gegevensset met zinsparafrases
De functie compute_metrics geeft accuracy en F1-score terug
Je hebt training_args in een eerdere oefening gedefinieerd

Deze oefening maakt deel uit van de cursus

Efficiënt AI-modellen trainen met PyTorch

Bekijk cursus

Oefeninstructies

Geef het model door aan de klasse Trainer().
Geef de trainingsargumenten door aan de klasse Trainer().
Geef een functie door aan de klasse Trainer() om de metrics te berekenen.
Print het apparaat dat trainer kiest.

Interactieve oefening met praktijkervaring

Probeer deze oefening door deze voorbeeldcode aan te vullen.

trainer = Trainer(
    # Pass in the model
    model=____,
    # Input the training arguments
    args=____,
    train_dataset=dataset["train"],
    eval_dataset=dataset["validation"],
    # Pass in a function to compute metrics
    compute_metrics=____,
)

trainer.train()

# Print the device that the trainer chooses
print(trainer.args.____)

Code bewerken en uitvoeren