CommencerCommencer gratuitement

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

Afficher le cours

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,
)
Modifier et exécuter le code