Skip to content

Pytorch codebase for Capturing label characteristics in VAEs

Notifications You must be signed in to change notification settings

thwjoy/ccvae_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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).

About

Pytorch codebase for Capturing label characteristics in VAEs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages