MCPcopy Create free account

hub / github.com/AntixK/PyTorch-VAE / types & classes

Types & classes53 in github.com/AntixK/PyTorch-VAE

⨍ Functions 334◇ Types & classes 53
↓ 2 callersClassMyCelebA
A work-around to address issues with pytorch's celebA dataset class. Download and Extract URL : https://drive.google.com/file/d/1m8-
dataset.py:27
↓ 2 callersClassResidualLayer
models/vq_vae.py:57
↓ 1 callersClassBetaTCVAE
models/betatc_vae.py:9
↓ 1 callersClassBetaVAE
models/beta_vae.py:8
↓ 1 callersClassDFCVAE
models/dfcvae.py:9
↓ 1 callersClassDIPVAE
models/dip_vae.py:8
↓ 1 callersClassEncoderBlock
models/lvae.py:12
↓ 1 callersClassFactorVAE
models/fvae.py:8
↓ 1 callersClassGammaVAE
models/gamma_vae.py:10
↓ 1 callersClassHVAE
models/hvae.py:8
↓ 1 callersClassIWAE
models/iwae.py:8
↓ 1 callersClassJointVAE
models/joint_vae.py:9
↓ 1 callersClassLVAE
models/lvae.py:62
↓ 1 callersClassLadderBlock
models/lvae.py:43
↓ 1 callersClassLogCoshVAE
models/logcosh_vae.py:8
↓ 1 callersClassMIWAE
models/miwae.py:9
↓ 1 callersClassMSSIM
models/mssim_vae.py:182
↓ 1 callersClassMSSIMVAE
models/mssim_vae.py:9
↓ 1 callersClassSWAE
models/swae.py:9
↓ 1 callersClassVAEDataset
PyTorch Lightning data module Args: data_dir: root directory of your dataset. train_batch_size: the batch size to use durin
dataset.py:66
↓ 1 callersClassVAEXperiment
experiment.py:15
↓ 1 callersClassVQVAE
models/vq_vae.py:73
↓ 1 callersClassVampVAE
models/vampvae.py:8
↓ 1 callersClassVanillaVAE
models/vanilla_vae.py:8
↓ 1 callersClassVectorQuantizer
Reference: [1] https://github.com/deepmind/sonnet/blob/v2/sonnet/src/nets/vqvae.py
models/vq_vae.py:7
↓ 1 callersClassWAE_MMD
models/wae_mmd.py:8
ClassBaseVAE
models/base.py:5
ClassCategoricalVAE
models/cat_vae.py:9
ClassConditionalVAE
models/cvae.py:8
ClassInfoVAE
models/info_vae.py:8
ClassMyDataset
dataset.py:15
ClassOxfordPets
URL = https://www.robots.ox.ac.uk/~vgg/data/pets/
dataset.py:40
ClassTestBetaTCVAE
tests/test_betatcvae.py:7
ClassTestCVAE
tests/text_cvae.py:6
ClassTestDFCVAE
tests/test_dfc.py:7
ClassTestDIPVAE
tests/test_dipvae.py:7
ClassTestFAE
tests/test_fvae.py:7
ClassTestGammaVAE
tests/test_gvae.py:7
ClassTestHVAE
tests/test_hvae.py:7
ClassTestIWAE
tests/test_iwae.py:7
ClassTestLVAE
tests/test_lvae.py:7
ClassTestMIWAE
tests/test_miwae.py:7
ClassTestMSSIMVAE
tests/test_mssimvae.py:7
ClassTestSWAE
tests/test_swae.py:7
ClassTestVAE
tests/bvae.py:7
ClassTestVAE
tests/test_joint_Vae.py:7
ClassTestVAE
tests/test_vae.py:7
ClassTestVAE
tests/test_logcosh.py:7
ClassTestVAE
tests/test_cat_vae.py:7
ClassTestVQVAE
tests/test_vq_vae.py:7
ClassTestVVAE
tests/text_vamp.py:7
ClassTestWAE
tests/test_wae.py:7
ClassTwoStageVAE
models/twostage_vae.py:8