Ajustement fin en CV : configuration du trainer
Vous avez préparé le jeu de données et adapté un modèle préentraîné aux nouvelles classes ; il est maintenant temps de configurer votre trainer.
Les TrainingArguments et Trainer ont été chargés depuis la bibliothèque transformers. Le modèle (model) et le jeu de données (dataset) ont été chargés selon votre configuration précédente.
Cet exercice fait partie du cours
Modèles multimodaux avec Hugging Face
Instructions
- Réglez le taux d’apprentissage sur
6e-5. - Fournissez le modèle, les données d’entraînement et les données de test à l’instance
Trainer.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
training_args = TrainingArguments(
output_dir="dataset_finetune",
# Adjust the learning rate
____,
gradient_accumulation_steps=4,
num_train_epochs=3,
push_to_hub=False
)
trainer = Trainer(
# Provide the model and datasets
model=____,
args=training_args,
data_collator=data_collator,
train_dataset=____,
eval_dataset=____,
processing_class=image_processor,
compute_metrics=compute_metrics,
)