Den Trainer einrichten

Dein Modell ersetzt komplexe Formulierungen durch einfachere, besser zugängliche Paraphrasen, um die Lesbarkeit von Übersetzungen zu verbessern. Es kann zum Beispiel 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 braucht durch den Aufruf trainer.train() etwas Zeit.

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 vorherigen Übung definiert

Diese Übung ist Teil des Kurses

Effizientes KI-Modelltraining mit PyTorch

Kurs anzeigen

Anleitung zur Übung

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

Interaktive Übung

Vervollständige den Beispielcode, um diese Übung erfolgreich abzuschließen.

trainer = Trainer(
    # Pass in the model
    ____=____,
    # Input the training arguments
    ____=____,
    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(____.args.____)

Code bearbeiten und ausführen