**Additional context** I would like to add a WGAN model trained on Celeb A dataset in pytorch.
Additional context
I would like to add a WGAN model trained on Celeb A dataset in pytorch.