Konfigurera Trainer

Din modell ska ersätta komplexa fraser med enklare, mer lättillgängliga omformuleringar för att förbättra läsbarheten hos översättningar. Till exempel kan den förenkla meningen "The inclement weather conditions precipitated the postponement of the outdoor event" till "The bad weather caused the outdoor event to be delayed." Bygg upp Trainer för att förbereda träningen av din språköversättningstjänst! Övningen tar en stund att köra när trainer.train() anropas.

En del data har lästs in i förväg:

model är en Transformer-modell
dataset innehåller MRPC-datamängden med meningsparafraser
Funktionen compute_metrics returnerar precision och F1-värde
Du har definierat training_args i en tidigare övning

Den här övningen är en del av kursen

Effektiv AI-modellträning med PyTorch

Visa kurs

Övningsinstruktioner

Skicka in model till klassen Trainer().
Skicka in träningsargumenten till klassen Trainer().
Skicka in en funktion för att beräkna mätvärden till klassen Trainer().
Skriv ut den enhet som trainer väljer.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

Redigera och kör kod