Open Thekey756 opened 4 months ago
Hi, I used a custom celebA conditioned dataset to train this model. This was done by using the following two attributes: 1) gender (male/female) and 2) Smiling/not-smiling. This is the reason you also see four classes in the celebA dataset. Feel free to construct any other variation of the celebA dataset for the conditional diffusion training.
Thank you for providing code, I'm trying to train the model on Celeba, I have downloaded the dataset and found it has no labels, however in the function get_metadata() in file data.py, code is written as below:
I dont know if you did some modification on Celeba, as the total dataset contains 200K+ images while you only use 109036 images as train_iamges, and the num_classes equals 4 is also my question.
In addition, I found the script used for training on Celeba:
In the Celeba I downloaded ,there are no folder named celebA_male_smile_64_balanced/train
Hoping for your reply, thanks in advance.