How to fine-tune on custom dataset?

mlfoundations / model-soups

Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference time

MIT License

426 stars 38 forks source link

How to fine-tune on custom dataset? #11

Open kenhuang1964 opened 1 year ago

kenhuang1964 commented 1 year ago

Hey guys, I have this classification dataset with currency denominations but I'm not sure how to use model-soup for fine-tuning. The way my dataset is structured is that I have several folders for each denomination and each folder has thousands of images. I have a train, valid, and test folder overall. Can someone help me with this please?

mitchellnw commented 1 year ago

The example finetune.py may be helpful -- it's for imagenet but I believe the structure is similar