├── .gitignore ├── .idea ├── .gitignore ├── PyTorch-VAE.iml ├── inspectionProfiles │ └── profiles_settings.xml ├── misc.xml ├── modules.xml └── vcs.xml ├── LICENSE.md ├── README.md ├── assets ├── BetaTCVAE_49.png ├── BetaVAE_B_20.png ├── BetaVAE_B_35.png ├── BetaVAE_H_20.png ├── CategoricalVAE_49.png ├── ConditionalVAE_20.png ├── DFCVAE_49.png ├── DIPVAE_83.png ├── IWAE_19.png ├── InfoVAE_31.png ├── InfoVAE_7.png ├── JointVAE_49.png ├── LogCoshVAE_49.png ├── MIWAE_29.png ├── MSSIMVAE_29.png ├── SWAE_49.png ├── Vanilla VAE_25.png ├── WAE_IMQ_15.png ├── WAE_RBF_18.png ├── recons_BetaTCVAE_49.png ├── recons_BetaVAE_B_20.png ├── recons_BetaVAE_B_35.png ├── recons_BetaVAE_H_20.png ├── recons_CategoricalVAE_49.png ├── recons_ConditionalVAE_20.png ├── recons_DFCVAE_49.png ├── recons_DIPVAE_83.png ├── recons_IWAE_19.png ├── recons_InfoVAE_31.png ├── recons_JointVAE_49.png ├── recons_LogCoshVAE_49.png ├── recons_MIWAE_29.png ├── recons_MSSIMVAE_29.png ├── recons_SWAE_49.png ├── recons_VQVAE_29.png ├── recons_Vanilla VAE_25.png ├── recons_WAE_IMQ_15.png └── recons_WAE_RBF_19.png ├── configs ├── bbvae.yaml ├── betatc_vae.yaml ├── bhvae.yaml ├── cat_vae.yaml ├── cvae.yaml ├── dfc_vae.yaml ├── dip_vae.yaml ├── factorvae.yaml ├── gammavae.yaml ├── hvae.yaml ├── infovae.yaml ├── iwae.yaml ├── joint_vae.yaml ├── logcosh_vae.yaml ├── lvae.yaml ├── miwae.yaml ├── mssim_vae.yaml ├── swae.yaml ├── vae.yaml ├── vampvae.yaml ├── vq_vae.yaml ├── wae_mmd_imq.yaml └── wae_mmd_rbf.yaml ├── dataset.py ├── experiment.py ├── models ├── __init__.py ├── base.py ├── beta_vae.py ├── betatc_vae.py ├── cat_vae.py ├── cvae.py ├── dfcvae.py ├── dip_vae.py ├── fvae.py ├── gamma_vae.py ├── hvae.py ├── info_vae.py ├── iwae.py ├── joint_vae.py ├── logcosh_vae.py ├── lvae.py ├── miwae.py ├── mssim_vae.py ├── swae.py ├── twostage_vae.py ├── types_.py ├── vampvae.py ├── vanilla_vae.py ├── vq_vae.py └── wae_mmd.py ├── requirements.txt ├── run.py ├── tests ├── bvae.py ├── test_betatcvae.py ├── test_cat_vae.py ├── test_dfc.py ├── test_dipvae.py ├── test_fvae.py ├── test_gvae.py ├── test_hvae.py ├── test_iwae.py ├── test_joint_Vae.py ├── test_logcosh.py ├── test_lvae.py ├── test_miwae.py ├── test_mssimvae.py ├── test_swae.py ├── test_vae.py ├── test_vq_vae.py ├── test_wae.py ├── text_cvae.py └── text_vamp.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /workspace.xml 3 | -------------------------------------------------------------------------------- /.idea/PyTorch-VAE.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.idea/PyTorch-VAE.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /.idea/vcs.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/.idea/vcs.xml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/README.md -------------------------------------------------------------------------------- /assets/BetaTCVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/BetaTCVAE_49.png -------------------------------------------------------------------------------- /assets/BetaVAE_B_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/BetaVAE_B_20.png -------------------------------------------------------------------------------- /assets/BetaVAE_B_35.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/BetaVAE_B_35.png -------------------------------------------------------------------------------- /assets/BetaVAE_H_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/BetaVAE_H_20.png -------------------------------------------------------------------------------- /assets/CategoricalVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/CategoricalVAE_49.png -------------------------------------------------------------------------------- /assets/ConditionalVAE_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/ConditionalVAE_20.png -------------------------------------------------------------------------------- /assets/DFCVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/DFCVAE_49.png -------------------------------------------------------------------------------- /assets/DIPVAE_83.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/DIPVAE_83.png -------------------------------------------------------------------------------- /assets/IWAE_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/IWAE_19.png -------------------------------------------------------------------------------- /assets/InfoVAE_31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/InfoVAE_31.png -------------------------------------------------------------------------------- /assets/InfoVAE_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/InfoVAE_7.png -------------------------------------------------------------------------------- /assets/JointVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/JointVAE_49.png -------------------------------------------------------------------------------- /assets/LogCoshVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/LogCoshVAE_49.png -------------------------------------------------------------------------------- /assets/MIWAE_29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/MIWAE_29.png -------------------------------------------------------------------------------- /assets/MSSIMVAE_29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/MSSIMVAE_29.png -------------------------------------------------------------------------------- /assets/SWAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/SWAE_49.png -------------------------------------------------------------------------------- /assets/Vanilla VAE_25.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/Vanilla VAE_25.png -------------------------------------------------------------------------------- /assets/WAE_IMQ_15.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/WAE_IMQ_15.png -------------------------------------------------------------------------------- /assets/WAE_RBF_18.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/WAE_RBF_18.png -------------------------------------------------------------------------------- /assets/recons_BetaTCVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_BetaTCVAE_49.png -------------------------------------------------------------------------------- /assets/recons_BetaVAE_B_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_BetaVAE_B_20.png -------------------------------------------------------------------------------- /assets/recons_BetaVAE_B_35.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_BetaVAE_B_35.png -------------------------------------------------------------------------------- /assets/recons_BetaVAE_H_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_BetaVAE_H_20.png -------------------------------------------------------------------------------- /assets/recons_CategoricalVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_CategoricalVAE_49.png -------------------------------------------------------------------------------- /assets/recons_ConditionalVAE_20.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_ConditionalVAE_20.png -------------------------------------------------------------------------------- /assets/recons_DFCVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_DFCVAE_49.png -------------------------------------------------------------------------------- /assets/recons_DIPVAE_83.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_DIPVAE_83.png -------------------------------------------------------------------------------- /assets/recons_IWAE_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_IWAE_19.png -------------------------------------------------------------------------------- /assets/recons_InfoVAE_31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_InfoVAE_31.png -------------------------------------------------------------------------------- /assets/recons_JointVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_JointVAE_49.png -------------------------------------------------------------------------------- /assets/recons_LogCoshVAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_LogCoshVAE_49.png -------------------------------------------------------------------------------- /assets/recons_MIWAE_29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_MIWAE_29.png -------------------------------------------------------------------------------- /assets/recons_MSSIMVAE_29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_MSSIMVAE_29.png -------------------------------------------------------------------------------- /assets/recons_SWAE_49.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_SWAE_49.png -------------------------------------------------------------------------------- /assets/recons_VQVAE_29.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_VQVAE_29.png -------------------------------------------------------------------------------- /assets/recons_Vanilla VAE_25.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_Vanilla VAE_25.png -------------------------------------------------------------------------------- /assets/recons_WAE_IMQ_15.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_WAE_IMQ_15.png -------------------------------------------------------------------------------- /assets/recons_WAE_RBF_19.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/assets/recons_WAE_RBF_19.png -------------------------------------------------------------------------------- /configs/bbvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/bbvae.yaml -------------------------------------------------------------------------------- /configs/betatc_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/betatc_vae.yaml -------------------------------------------------------------------------------- /configs/bhvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/bhvae.yaml -------------------------------------------------------------------------------- /configs/cat_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/cat_vae.yaml -------------------------------------------------------------------------------- /configs/cvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/cvae.yaml -------------------------------------------------------------------------------- /configs/dfc_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/dfc_vae.yaml -------------------------------------------------------------------------------- /configs/dip_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/dip_vae.yaml -------------------------------------------------------------------------------- /configs/factorvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/factorvae.yaml -------------------------------------------------------------------------------- /configs/gammavae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/gammavae.yaml -------------------------------------------------------------------------------- /configs/hvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/hvae.yaml -------------------------------------------------------------------------------- /configs/infovae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/infovae.yaml -------------------------------------------------------------------------------- /configs/iwae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/iwae.yaml -------------------------------------------------------------------------------- /configs/joint_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/joint_vae.yaml -------------------------------------------------------------------------------- /configs/logcosh_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/logcosh_vae.yaml -------------------------------------------------------------------------------- /configs/lvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/lvae.yaml -------------------------------------------------------------------------------- /configs/miwae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/miwae.yaml -------------------------------------------------------------------------------- /configs/mssim_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/mssim_vae.yaml -------------------------------------------------------------------------------- /configs/swae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/swae.yaml -------------------------------------------------------------------------------- /configs/vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/vae.yaml -------------------------------------------------------------------------------- /configs/vampvae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/vampvae.yaml -------------------------------------------------------------------------------- /configs/vq_vae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/vq_vae.yaml -------------------------------------------------------------------------------- /configs/wae_mmd_imq.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/wae_mmd_imq.yaml -------------------------------------------------------------------------------- /configs/wae_mmd_rbf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/configs/wae_mmd_rbf.yaml -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/dataset.py -------------------------------------------------------------------------------- /experiment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/experiment.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/base.py -------------------------------------------------------------------------------- /models/beta_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/beta_vae.py -------------------------------------------------------------------------------- /models/betatc_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/betatc_vae.py -------------------------------------------------------------------------------- /models/cat_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/cat_vae.py -------------------------------------------------------------------------------- /models/cvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/cvae.py -------------------------------------------------------------------------------- /models/dfcvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/dfcvae.py -------------------------------------------------------------------------------- /models/dip_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/dip_vae.py -------------------------------------------------------------------------------- /models/fvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/fvae.py -------------------------------------------------------------------------------- /models/gamma_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/gamma_vae.py -------------------------------------------------------------------------------- /models/hvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/hvae.py -------------------------------------------------------------------------------- /models/info_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/info_vae.py -------------------------------------------------------------------------------- /models/iwae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/iwae.py -------------------------------------------------------------------------------- /models/joint_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/joint_vae.py -------------------------------------------------------------------------------- /models/logcosh_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/logcosh_vae.py -------------------------------------------------------------------------------- /models/lvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/lvae.py -------------------------------------------------------------------------------- /models/miwae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/miwae.py -------------------------------------------------------------------------------- /models/mssim_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/mssim_vae.py -------------------------------------------------------------------------------- /models/swae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/swae.py -------------------------------------------------------------------------------- /models/twostage_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/twostage_vae.py -------------------------------------------------------------------------------- /models/types_.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/types_.py -------------------------------------------------------------------------------- /models/vampvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/vampvae.py -------------------------------------------------------------------------------- /models/vanilla_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/vanilla_vae.py -------------------------------------------------------------------------------- /models/vq_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/vq_vae.py -------------------------------------------------------------------------------- /models/wae_mmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/models/wae_mmd.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/run.py -------------------------------------------------------------------------------- /tests/bvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/bvae.py -------------------------------------------------------------------------------- /tests/test_betatcvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_betatcvae.py -------------------------------------------------------------------------------- /tests/test_cat_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_cat_vae.py -------------------------------------------------------------------------------- /tests/test_dfc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_dfc.py -------------------------------------------------------------------------------- /tests/test_dipvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_dipvae.py -------------------------------------------------------------------------------- /tests/test_fvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_fvae.py -------------------------------------------------------------------------------- /tests/test_gvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_gvae.py -------------------------------------------------------------------------------- /tests/test_hvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_hvae.py -------------------------------------------------------------------------------- /tests/test_iwae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_iwae.py -------------------------------------------------------------------------------- /tests/test_joint_Vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_joint_Vae.py -------------------------------------------------------------------------------- /tests/test_logcosh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_logcosh.py -------------------------------------------------------------------------------- /tests/test_lvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_lvae.py -------------------------------------------------------------------------------- /tests/test_miwae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_miwae.py -------------------------------------------------------------------------------- /tests/test_mssimvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_mssimvae.py -------------------------------------------------------------------------------- /tests/test_swae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_swae.py -------------------------------------------------------------------------------- /tests/test_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_vae.py -------------------------------------------------------------------------------- /tests/test_vq_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_vq_vae.py -------------------------------------------------------------------------------- /tests/test_wae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/test_wae.py -------------------------------------------------------------------------------- /tests/text_cvae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/text_cvae.py -------------------------------------------------------------------------------- /tests/text_vamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/tests/text_vamp.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AntixK/PyTorch-VAE/HEAD/utils.py --------------------------------------------------------------------------------