New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

customized dataset #307

Open

zhuang-li opened this issue Sep 3, 2022 · 1 comment

zhuang-li commented Sep 3, 2022 •

edited

Loading

Hi sorry for this naive question as I am new to computer vision and DALLE. Say if I have a folder of images and captions such as

dir/
cat.jpg
cat.txt
dog.jpg
dog.txt

how to convert the folder into the data format the DALLE-mini can accept if I want to train a new model? Is there any script in this repo?

I was using "https://github.com/lucidrains/DALLE-pytorch" the data format for their model is the above format but I am not clear what is the data format used here.

Owner

borisdayma commented Sep 4, 2022

This is the required format: https://github.com/borisdayma/dalle-mini/blob/main/tools/dataset/encode_dataset.ipynb

Umal-exvc mentioned this issue

How to retrain(fine-tuned) custom datasets #317

Open

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment