├── .ipynb_checkpoints ├── main-checkpoint.ipynb └── readme-checkpoint.ipynb ├── FMNISTDataset.py ├── README.md ├── TbLogger.py ├── Util.py ├── VAE.py ├── VAELoss.py ├── baseline_train.py ├── compare.png ├── dockerfile └── Dockerfile ├── extensions.png ├── main.ipynb ├── pytorch_ssim ├── __init__.py └── __pycache__ │ ├── __init__.cpython-35.pyc │ └── __init__.cpython-36.pyc ├── readme.ipynb ├── reconstructions.png ├── results ├── reconstruction_1.png ├── reconstruction_10.png ├── reconstruction_2.png ├── reconstruction_3.png ├── reconstruction_4.png ├── reconstruction_5.png ├── reconstruction_6.png ├── reconstruction_7.png ├── reconstruction_8.png └── reconstruction_9.png ├── s.sh ├── tb_logs ├── .gitignore └── fmnist_fc_conv_l10_rebalance_no_norm │ └── events.out.tfevents.1530116723.130a2f3b6351 ├── train.py └── weights ├── .gitignore ├── fmnist_fc_conv_l10_rebalance_no_norm_best.pth.tar └── fmnist_fc_conv_l10_rebalance_no_norm_checkpoint.pth.tar /.ipynb_checkpoints/main-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/.ipynb_checkpoints/main-checkpoint.ipynb -------------------------------------------------------------------------------- /.ipynb_checkpoints/readme-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/.ipynb_checkpoints/readme-checkpoint.ipynb -------------------------------------------------------------------------------- /FMNISTDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/FMNISTDataset.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/README.md -------------------------------------------------------------------------------- /TbLogger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/TbLogger.py -------------------------------------------------------------------------------- /Util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/Util.py -------------------------------------------------------------------------------- /VAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/VAE.py -------------------------------------------------------------------------------- /VAELoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/VAELoss.py -------------------------------------------------------------------------------- /baseline_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/baseline_train.py -------------------------------------------------------------------------------- /compare.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/compare.png -------------------------------------------------------------------------------- /dockerfile/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/dockerfile/Dockerfile -------------------------------------------------------------------------------- /extensions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/extensions.png -------------------------------------------------------------------------------- /main.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/main.ipynb -------------------------------------------------------------------------------- /pytorch_ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/pytorch_ssim/__init__.py -------------------------------------------------------------------------------- /pytorch_ssim/__pycache__/__init__.cpython-35.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/pytorch_ssim/__pycache__/__init__.cpython-35.pyc -------------------------------------------------------------------------------- /pytorch_ssim/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/pytorch_ssim/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /readme.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/readme.ipynb -------------------------------------------------------------------------------- /reconstructions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/reconstructions.png -------------------------------------------------------------------------------- /results/reconstruction_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_1.png -------------------------------------------------------------------------------- /results/reconstruction_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_10.png -------------------------------------------------------------------------------- /results/reconstruction_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_2.png -------------------------------------------------------------------------------- /results/reconstruction_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_3.png -------------------------------------------------------------------------------- /results/reconstruction_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_4.png -------------------------------------------------------------------------------- /results/reconstruction_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_5.png -------------------------------------------------------------------------------- /results/reconstruction_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_6.png -------------------------------------------------------------------------------- /results/reconstruction_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_7.png -------------------------------------------------------------------------------- /results/reconstruction_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_8.png -------------------------------------------------------------------------------- /results/reconstruction_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/results/reconstruction_9.png -------------------------------------------------------------------------------- /s.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/s.sh -------------------------------------------------------------------------------- /tb_logs/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/tb_logs/.gitignore -------------------------------------------------------------------------------- /tb_logs/fmnist_fc_conv_l10_rebalance_no_norm/events.out.tfevents.1530116723.130a2f3b6351: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/tb_logs/fmnist_fc_conv_l10_rebalance_no_norm/events.out.tfevents.1530116723.130a2f3b6351 -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/train.py -------------------------------------------------------------------------------- /weights/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/weights/.gitignore -------------------------------------------------------------------------------- /weights/fmnist_fc_conv_l10_rebalance_no_norm_best.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/weights/fmnist_fc_conv_l10_rebalance_no_norm_best.pth.tar -------------------------------------------------------------------------------- /weights/fmnist_fc_conv_l10_rebalance_no_norm_checkpoint.pth.tar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/snakers4/playing_with_vae/HEAD/weights/fmnist_fc_conv_l10_rebalance_no_norm_checkpoint.pth.tar --------------------------------------------------------------------------------