Fine-tune with a custom dataset
iremonur opened this issue · comments
Hi,
I try to fine tune the clip with a very small custom dataset. I ran the command "python train_finetune.py --folder /home/ionur2/train-CLIP/data --batch_size 2" and I get an assertion error :
Is there anyone facing the same issue?
I would be glad if you help me with this!
Hi, I solved the issue thanks to @ymzhu19eee in the issue #20