1. Nauka
  2. /
  3. Kursy
  4. /
  5. Modele multimodalne z Hugging Face

Connected

ćwiczenie

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ą.

Instrukcje

100 XP
  • Ustaw współczynnik uczenia na 6e-5.
  • Przekaż model, dane treningowe i dane testowe do instancji Trainer.