├── .gitignore ├── README.md ├── atari ├── README.md ├── images │ ├── alt.png │ └── cvae.png ├── main.lua ├── models │ ├── Cnn.lua │ ├── CnnCVae1.lua │ ├── CnnCVae2.lua │ ├── CnnCVae3.lua │ ├── README.md │ └── StackedCnn.lua ├── nnutils │ ├── FixedIndex.lua │ ├── GaussianCriterion.lua │ ├── KLDCriterion.lua │ ├── README.md │ ├── Sampler.lua │ ├── SplitTensor.lua │ └── init.lua └── utils │ ├── DataLoader.lua │ ├── DataManager.lua │ ├── Logger.lua │ ├── README.md │ ├── Saver.lua │ └── init.lua └── modality ├── README.md ├── datasets └── mnist.hdf5 ├── images ├── alt.png ├── frey.gif ├── gmm-characteristic-z10-1000.png ├── gmm-encoding-1000.png ├── gmm-encoding.png ├── gmm-manifold.png ├── gmm-prior-1000.png ├── gmm-prior.png ├── gmm-z10-1000.png ├── gmm-z10.png ├── indgmm-encoding-1000.png ├── indgmm-encoding.png ├── indgmm-manifold.png ├── indgmm-prior-1000.png ├── indgmm-prior.png ├── indgmm-z10-1000.png ├── indgmm-z10.png ├── mnist.gif ├── sample.png ├── vae-encoding.png ├── vae-manifold.png ├── vae-prior.png ├── vae-z10.png ├── vae.png ├── z10.png └── z2-10.png ├── misc └── gmm_kld_derivation.pdf ├── mnist.lua ├── models ├── GmmVae.lua ├── README.md ├── Vae.lua └── init.lua ├── nnutils ├── GaussianCriterion.lua ├── GmmKLDCriterion.lua ├── GmmSampler.lua ├── KLDCriterion.lua ├── README.md ├── Sampler.lua ├── SplitTensor.lua ├── Zero.lua └── init.lua └── utils ├── DataManager.lua ├── Logger.lua ├── MnistManager.lua ├── README.md └── grid.lua /.gitignore: -------------------------------------------------------------------------------- 1 | save 2 | .ipynb* 3 | .dropbox* -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/README.md -------------------------------------------------------------------------------- /atari/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/README.md -------------------------------------------------------------------------------- /atari/images/alt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/images/alt.png -------------------------------------------------------------------------------- /atari/images/cvae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/images/cvae.png -------------------------------------------------------------------------------- /atari/main.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/main.lua -------------------------------------------------------------------------------- /atari/models/Cnn.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/Cnn.lua -------------------------------------------------------------------------------- /atari/models/CnnCVae1.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/CnnCVae1.lua -------------------------------------------------------------------------------- /atari/models/CnnCVae2.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/CnnCVae2.lua -------------------------------------------------------------------------------- /atari/models/CnnCVae3.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/CnnCVae3.lua -------------------------------------------------------------------------------- /atari/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/README.md -------------------------------------------------------------------------------- /atari/models/StackedCnn.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/models/StackedCnn.lua -------------------------------------------------------------------------------- /atari/nnutils/FixedIndex.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/FixedIndex.lua -------------------------------------------------------------------------------- /atari/nnutils/GaussianCriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/GaussianCriterion.lua -------------------------------------------------------------------------------- /atari/nnutils/KLDCriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/KLDCriterion.lua -------------------------------------------------------------------------------- /atari/nnutils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/README.md -------------------------------------------------------------------------------- /atari/nnutils/Sampler.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/Sampler.lua -------------------------------------------------------------------------------- /atari/nnutils/SplitTensor.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/SplitTensor.lua -------------------------------------------------------------------------------- /atari/nnutils/init.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/nnutils/init.lua -------------------------------------------------------------------------------- /atari/utils/DataLoader.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/DataLoader.lua -------------------------------------------------------------------------------- /atari/utils/DataManager.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/DataManager.lua -------------------------------------------------------------------------------- /atari/utils/Logger.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/Logger.lua -------------------------------------------------------------------------------- /atari/utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/README.md -------------------------------------------------------------------------------- /atari/utils/Saver.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/Saver.lua -------------------------------------------------------------------------------- /atari/utils/init.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/atari/utils/init.lua -------------------------------------------------------------------------------- /modality/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/README.md -------------------------------------------------------------------------------- /modality/datasets/mnist.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/datasets/mnist.hdf5 -------------------------------------------------------------------------------- /modality/images/alt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/alt.png -------------------------------------------------------------------------------- /modality/images/frey.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/frey.gif -------------------------------------------------------------------------------- /modality/images/gmm-characteristic-z10-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-characteristic-z10-1000.png -------------------------------------------------------------------------------- /modality/images/gmm-encoding-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-encoding-1000.png -------------------------------------------------------------------------------- /modality/images/gmm-encoding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-encoding.png -------------------------------------------------------------------------------- /modality/images/gmm-manifold.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-manifold.png -------------------------------------------------------------------------------- /modality/images/gmm-prior-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-prior-1000.png -------------------------------------------------------------------------------- /modality/images/gmm-prior.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-prior.png -------------------------------------------------------------------------------- /modality/images/gmm-z10-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-z10-1000.png -------------------------------------------------------------------------------- /modality/images/gmm-z10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/gmm-z10.png -------------------------------------------------------------------------------- /modality/images/indgmm-encoding-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-encoding-1000.png -------------------------------------------------------------------------------- /modality/images/indgmm-encoding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-encoding.png -------------------------------------------------------------------------------- /modality/images/indgmm-manifold.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-manifold.png -------------------------------------------------------------------------------- /modality/images/indgmm-prior-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-prior-1000.png -------------------------------------------------------------------------------- /modality/images/indgmm-prior.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-prior.png -------------------------------------------------------------------------------- /modality/images/indgmm-z10-1000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-z10-1000.png -------------------------------------------------------------------------------- /modality/images/indgmm-z10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/indgmm-z10.png -------------------------------------------------------------------------------- /modality/images/mnist.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/mnist.gif -------------------------------------------------------------------------------- /modality/images/sample.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/sample.png -------------------------------------------------------------------------------- /modality/images/vae-encoding.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/vae-encoding.png -------------------------------------------------------------------------------- /modality/images/vae-manifold.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/vae-manifold.png -------------------------------------------------------------------------------- /modality/images/vae-prior.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/vae-prior.png -------------------------------------------------------------------------------- /modality/images/vae-z10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/vae-z10.png -------------------------------------------------------------------------------- /modality/images/vae.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/vae.png -------------------------------------------------------------------------------- /modality/images/z10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/z10.png -------------------------------------------------------------------------------- /modality/images/z2-10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/images/z2-10.png -------------------------------------------------------------------------------- /modality/misc/gmm_kld_derivation.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/misc/gmm_kld_derivation.pdf -------------------------------------------------------------------------------- /modality/mnist.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/mnist.lua -------------------------------------------------------------------------------- /modality/models/GmmVae.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/models/GmmVae.lua -------------------------------------------------------------------------------- /modality/models/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/models/README.md -------------------------------------------------------------------------------- /modality/models/Vae.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/models/Vae.lua -------------------------------------------------------------------------------- /modality/models/init.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/models/init.lua -------------------------------------------------------------------------------- /modality/nnutils/GaussianCriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/GaussianCriterion.lua -------------------------------------------------------------------------------- /modality/nnutils/GmmKLDCriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/GmmKLDCriterion.lua -------------------------------------------------------------------------------- /modality/nnutils/GmmSampler.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/GmmSampler.lua -------------------------------------------------------------------------------- /modality/nnutils/KLDCriterion.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/KLDCriterion.lua -------------------------------------------------------------------------------- /modality/nnutils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/README.md -------------------------------------------------------------------------------- /modality/nnutils/Sampler.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/Sampler.lua -------------------------------------------------------------------------------- /modality/nnutils/SplitTensor.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/SplitTensor.lua -------------------------------------------------------------------------------- /modality/nnutils/Zero.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/Zero.lua -------------------------------------------------------------------------------- /modality/nnutils/init.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/nnutils/init.lua -------------------------------------------------------------------------------- /modality/utils/DataManager.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/utils/DataManager.lua -------------------------------------------------------------------------------- /modality/utils/Logger.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/utils/Logger.lua -------------------------------------------------------------------------------- /modality/utils/MnistManager.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/utils/MnistManager.lua -------------------------------------------------------------------------------- /modality/utils/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/utils/README.md -------------------------------------------------------------------------------- /modality/utils/grid.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/RuiShu/vae-experiments/HEAD/modality/utils/grid.lua --------------------------------------------------------------------------------