1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Multi-Modal Models with Hugging Face
Connected

cvičení

Fine-tuning CV: konfigurace tréneru

Teď, když máš připravený dataset a předtrénovaný model upravený pro nové třídy, je čas nakonfigurovat trenér.

TrainingArguments a Trainer byly načteny z knihovny transformers. Model (model) a dataset (dataset) jsou načteny tak, jak jsi je dříve nastavil/a.

Pokyny

100 XP
  • Nastav learning rate na 6e-5.
  • Předej instanci Trainer model, trénovací data a testovací data.