Skip to content

amberyzheng/vae-cvae-manifold-dim

Repository files navigation

Learning Manifold Dimensions with Conditional Variational Autoencoders

Implementation in PyTorch

Dependencies

  • torch==1.10.2
  • torchvision==0.11.3
  • pyro==1.8.0
  • numpy==1.19.5

Running the code

  • Table 1: python3 sync_cvae_std.py
  • Table 2: python3 mnist_vae.py
  • Table 3: python3 sync_cvae_param
  • Table 4: python3 mnist_cvae.py
  • Table 5: python3 sync_adaptive_discrete.py
  • Table 6: python3 sync_adaptive_continuous.py
  • Table 7: First column is from sync_cvae_std.py. For second and third columns run python3 sync_decoder_structure.py
  • Table 8: python3 sync_sequence.py

About

[NeurIPS 2022] Learning Manifold Dimensions with Conditional Variational Autoencoders

Resources

Stars

11 stars

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages