1. Learn
    2. /
  2. Cursuri
    3. /
  3. Modele Multi-Modale cu Hugging Face
Connected

exercițiu

Fine-tuning CV: clasele modelului

În acest exercițiu, vei încărca modelul pre-antrenat și vei adapta ieșirea pentru a acomoda o nouă clasificare a tipurilor de mașini din setul de date Stanford Cars, în locul celor 1.000 de clase folosite la antrenarea originală pe ImageNet. Setul de date conține imagini etichetate cu mașini.

Setul de date a fost încărcat (dataset), la fel și AutoModelForImageClassification din transformers. Setul de date a fost filtrat astfel încât să includă trei tipuri de modele.

Instrucțiuni

100 XP
  • Obține noile nume de etichete din setul de date.
  • Adaugă noul mapping id2label la încărcarea modelului.
  • Adaugă mapping-ul corespunzător label2id.
  • Adaugă indicatorul necesar pentru a modifica numărul de clase.