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

cvičení

Fine-tuning CV modelu: třídy modelu

V tomto cvičení načteš předtrénovaný model a upravíš jeho výstup tak, aby odpovídal nové klasifikaci typů automobilů z datasetu Stanford Cars – místo původních 1 000 tříd použitých při trénování na ImageNetu. Dataset obsahuje obrázky aut s popisky.

Dataset je už načtený (dataset), stejně jako AutoModelForImageClassification z transformers. Dataset byl filtrovaný tak, aby obsahoval tři typy automobilů.

Pokyny

100 XP
  • Získej nové názvy tříd z datasetu.
  • Přidej nové mapování id2label při načítání modelu.
  • Přidej odpovídající mapování label2id.
  • Přidej potřebný příznak pro změnu počtu tříd.