model-soups
model-soups copied to clipboard
How to fine-tune on custom dataset?
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?
The example finetune.py may be helpful -- it's for imagenet but I believe the structure is similar