Configurer le Trainer

Votre modèle remplacera des expressions complexes par des reformulations plus simples et accessibles afin d’améliorer la lisibilité des traductions. Par exemple, il peut simplifier la phrase originale « The inclement weather conditions precipitated the postponement of the outdoor event » par « The bad weather caused the outdoor event to be delayed. » Préparez le Trainer pour entraîner votre service de traduction linguistique ! L’exécution de l’exercice prendra un certain temps avec l’appel à trainer.train().

Certaines données ont été préchargées :

model est un modèle Transformer
dataset contient l’ensemble de données MRPC de paraphrases de phrases
La fonction compute_metrics retourne l’exactitude et la mesure F1
Vous avez défini training_args dans un exercice précédent

Cette activité fait partie du cours

Entraînement efficace de modèles d’IA avec PyTorch

Voir le cours

Instructions de l’exercice

Transmettez model à la classe Trainer().
Fournissez les paramètres d’entraînement dans la classe Trainer().
Transmettez une fonction de calcul des mesures à la classe Trainer().
Affichez l’appareil choisi par le trainer.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

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

Modifier et exécuter le code