Fine-tune with a custom dataset

Zasder3 / train-CLIP

A PyTorch Lightning solution to training OpenAI's CLIP from scratch.

MIT License

653 stars 78 forks source link

Fine-tune with a custom dataset #23

Closed iremonur closed 2 years ago

iremonur commented 2 years ago

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 : Screenshot from 2021-11-23 12-54-24 Is there anyone facing the same issue? I would be glad if you help me with this!

iremonur commented 2 years ago

Hi, I solved the issue thanks to @ymzhu19eee in the issue #20