Pytorch implementation of Semi-Supervised Disentanglement of Class-Related and Class-Independent Factors in VAE (PartedVAE).
This repository's structure is based on the joint-vae repository.
Use main.py to train the model. Add needed tests and evaluations at the end.
If you find our work useful in your research, please cite using:
@article{hajimiri2021semi,
title={Semi-Supervised Disentanglement of Class-Related and Class-Independent Factors in VAE},
author={Hajimiri, Sina and Lotfi, Aryo and Soleymani Baghshah, Mahdieh},
journal={arXiv preprint arXiv:2102.00892},
year={2021}
}
MIT