CommencerCommencer gratuitement

Ajustement du CV : configuration du formateur

Maintenant que vous avez préparé l'ensemble de données et adapté un modèle pré-entraîné aux nouvelles classes, il est temps de configurer votre entraîneur.

Les classes TrainingArguments et Trainer ont été chargées à partir de la bibliothèque transformers. Le modèle (model) et l'ensemble de données (dataset) ont été chargés conformément à votre configuration précédente.

Cet exercice fait partie du cours

Modèles multimodaux avec Hugging Face

Afficher le cours

Instructions

  • Veuillez ajuster le taux d'apprentissage à l'6e-5.
  • Veuillez fournir le modèle, les données d'entraînement et les données de test à l'instance d'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,
)
Modifier et exécuter le code