How to add even more cateogries to the model?

#5
by KushBor - opened

I want to add more categories — say expand the classification from 3 to 6 classes.
How can I change the classification head to support more labels?
What’s the best practice to fine-tune the expanded model on new data (with more classes)?

Thanks in advance.

Sign up or log in to comment