/EBGAN-CelebA

Pytorch implementation of Energy Based GAN for CelebA dataset

Primary LanguageJupyter Notebook

EBGAN-CelebA

Pytorch implementation of Energy Based GAN for CelebA dataset To run:

python celebA_EBGAN.py --dataroot [datapath] --dataset [celebA] [--cuda]

Add --cuda for GPU run(max no. of GPUs used = 1)