Dostrajanie CV: konfiguracja trenera
Zbiór danych jest już przygotowany, a pretrenowany model dostosowany do nowych klas. Czas skonfigurować trenera.
TrainingArguments i Trainer zostały wczytane z biblioteki transformers. Model (model) oraz zbiór danych (dataset) są załadowane zgodnie z wcześniejszą konfiguracją.
To ćwiczenie jest częścią kursu
Modele multimodalne z Hugging Face
Instrukcje do ćwiczenia
- Ustaw współczynnik uczenia na
6e-5. - Przekaż model, dane treningowe i dane testowe do instancji
Trainer.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
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,
)