thwjoy / ccvae_pytorch

Pytorch codebase for Capturing label characteristics in VAEs

Capturing label characteristics in VAEs

Pytorch repository for Capturing Label Chacteristics in VAEs ICLR 2021. We kindly ask that you cite our work if you plan to use this codebase:

@inproceedings{
    joy2021capturing,
    title={Capturing Label Characteristics in {\{}VAE{\}}s},
    author={Tom Joy and Sebastian Schmon and Philip Torr and Siddharth N and Tom Rainforth},
    booktitle={International Conference on Learning Representations},
    year={2021},
    url={https://openreview.net/forum?id=wQRlSUZ5V7B}
}

Usage

Ensure that CelebA is in the directory data/datasets/celeba, such that the path data/datasets/celeba/celeba/img_align_celeba/* is accessable.

To train, run:

python ss_vay.py -sup <sup-frac> --cuda>

where <sup-frac> is the fraction of supervised data (e.g. 0.004, 0.06, 0.2, 1.0).