model-soups icon indicating copy to clipboard operation
model-soups copied to clipboard

How to fine-tune on custom dataset?

Open kenhuang1964 opened this issue 1 year ago • 1 comments

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?

kenhuang1964 avatar Jul 31 '23 01:07 kenhuang1964

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

mitchellnw avatar Jul 31 '23 01:07 mitchellnw