Qustions about CelebA dataset?

yilundu / improved_contrastive_divergence

[ICML'21] Improved Contrastive Divergence Training of Energy Based Models

62 stars 14 forks source link

Qustions about CelebA dataset? #4

Open HeimingX opened 3 years ago

HeimingX commented 3 years ago

Hi,

Thanks for the impressive idea and high-quality code.

I am trying to experiment on the compositional learning with CelebA dataset. However, I just found the path for CelebA is defined in this line, it seems to be an npy file. I wonder where can I find the corresponding files for training? Thanks a lot.

yilundu commented 3 years ago

Hi,

Thanks for the question! You will need to download the CelebA-HQ dataset. Instead of loading the CelebA numpy files, you can then directly read the images. Feel free to let me know if there are any issue.