Nastavení Traineru

Tvůj model bude nahrazovat složité fráze jednoduššími a srozumitelnějšími parafrázemi, čímž zlepší čitelnost překladů. Například původní větu „The inclement weather conditions precipitated the postponement of the outdoor event" dokáže zjednodušit na „The bad weather caused the outdoor event to be delayed." Sestav Trainer a připrav model na trénování překladatelské služby! Volání trainer.train() může chvíli trvat.

Některá data jsou předpřipravena:

model je Transformer model
dataset obsahuje dataset MRPC s větami a jejich parafrázemi
funkce compute_metrics vrací přesnost (accuracy) a skóre F1
training_args jsi definoval/a v předchozím cvičení

Toto cvičení je součástí kurzu

Efficient AI Model Training with PyTorch

Zobrazit kurz

Pokyny k cvičení

Předej model do třídy Trainer().
Předej trénovací argumenty do třídy Trainer().
Předej funkci pro výpočet metrik do třídy Trainer().
Vypiš zařízení, které trainer zvolí.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

Upravit a spustit kód