Configurează Trainer-ul

Modelul tău va înlocui expresii complexe cu parafrazări mai simple și mai accesibile, pentru a îmbunătăți lizibilitatea traducerilor. De exemplu, poate simplifica expresia originală "The inclement weather conditions precipitated the postponement of the outdoor event" în "The bad weather caused the outdoor event to be delayed." Construiește Trainer-ul pentru a pregăti antrenamentul serviciului tău de traducere a limbajului! Exercițiul va dura ceva timp la rulare, din cauza apelului trainer.train().

Unele date sunt preîncărcate:

model este un model Transformer
dataset conține setul de date MRPC cu parafrazări de propoziții
funcția compute_metrics returnează acuratețea și scorul F1
ai definit training_args într-un exercițiu anterior

Acest exercițiu face parte din cursul

Antrenament eficient al modelelor AI cu PyTorch

Vezi cursul

Instrucțiuni pentru exercițiu

Transmite model clasei Trainer().
Introdu argumentele de antrenament în clasa Trainer().
Transmite o funcție pentru calculul metricilor clasei Trainer().
Afișează dispozitivul ales de trainer.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

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

Editează și rulează codul