Den Trainer einrichten

Dein Modell ersetzt komplexe Formulierungen durch einfachere, besser verständliche Paraphrasen, um die Lesbarkeit von Übersetzungen zu verbessern. Zum Beispiel kann es den ursprünglichen Satz "The inclement weather conditions precipitated the postponement of the outdoor event" durch "The bad weather caused the outdoor event to be delayed." vereinfachen. Baue den Trainer, um das Training deines Sprachübersetzungsdienstes vorzubereiten! Die Übung benötigt etwas Zeit, wenn trainer.train() aufgerufen wird.

Einige Daten wurden vorab geladen:

model ist ein Transformer-Modell
dataset enthält den MRPC-Datensatz mit Satzparaphrasen
Die Funktion compute_metrics gibt Accuracy und F1-Score zurück
Du hast training_args in einer früheren Übung definiert

Diese Übung ist Teil des Kurses

<Kurs>Effizientes KI-Modelltraining mit PyTorch</Kurs>

Kurs ansehen

Übungsanweisungen

Übergib das model an die Klasse Trainer().
Übergib die Trainingsargumente in der Klasse Trainer().
Übergib eine Funktion zur Metrikberechnung an die Klasse Trainer().
Gib das Gerät aus, das der trainer auswählt.

Interaktive praktische Übung

Versuche dich an dieser Übung, indem du diesen Beispielcode vervollständigst.

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 bearbeiten und ausführen