├── .gitignore ├── LICENSE ├── README.md ├── data_scripts ├── download_and_preprocess.sh ├── preprocess_celeba64.py ├── random_split.py └── utility_resize.py ├── docker_run.sh ├── efficient_vdvae_jax ├── __init__.py ├── data │ ├── __init__.py │ ├── cifar10_data_loader.py │ ├── generic_data_loader.py │ ├── imagenet_data_loader.py │ └── mnist_data_loader.py ├── hparams.cfg ├── model │ ├── __init__.py │ ├── adamax.py │ ├── autoencoder.py │ ├── conv2d.py │ ├── div_stats_utils.py │ ├── latent_layers.py │ ├── layers.py │ ├── losses.py │ ├── model.py │ ├── optimizers.py │ ├── schedules.py │ └── ssim.py ├── reset.sh ├── synthesize.py ├── train.py └── utils │ ├── __init__.py │ ├── denormalizer.py │ ├── ema_train_state.py │ ├── inference_helpers.py │ ├── normalizer.py │ ├── temperature_functions.py │ ├── train_helpers.py │ └── utils.py ├── efficient_vdvae_torch ├── __init__.py ├── data │ ├── __init__.py │ ├── cifar10_data_loader.py │ ├── generic_data_loader.py │ ├── imagenet_data_loader.py │ └── mnist_data_loader.py ├── hparams.cfg ├── model │ ├── __init__.py │ ├── adamax.py │ ├── autoencoder.py │ ├── conv2d.py │ ├── def_model.py │ ├── div_stats_utils.py │ ├── latent_layers.py │ ├── layers.py │ ├── losses.py │ ├── model.py │ ├── schedules.py │ └── ssim.py ├── reset.sh ├── synthesize.py ├── train.py ├── train.sh └── utils │ ├── __init__.py │ ├── temperature_functions.py │ └── utils.py ├── egs ├── binarized_mnist_baseline.cfg ├── celebA64_baseline.cfg ├── celebA64_compact.cfg ├── celebAHQ1024_baseline.cfg ├── celebAHQ256_5bits_baseline.cfg ├── celebAHQ256_8bits_baseline.cfg ├── celebAHQ256_8bits_compact.cfg ├── cifar10_C1.cfg ├── cifar10_baseline.cfg ├── ffhq1024_baseline.cfg ├── ffhq256_5bits_C1.cfg ├── ffhq256_5bits_baseline.cfg ├── ffhq256_8bits_baseline.cfg ├── ffhq256_8bits_compact.cfg ├── imagenet32_C1.cfg ├── imagenet32_baseline.cfg ├── imagenet32_compact.cfg ├── imagenet64_C1.cfg ├── imagenet64_baseline.cfg └── imagenet64_compact.cfg └── images ├── .DS_Store └── unconditional_samples.png /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/README.md -------------------------------------------------------------------------------- /data_scripts/download_and_preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/data_scripts/download_and_preprocess.sh -------------------------------------------------------------------------------- /data_scripts/preprocess_celeba64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/data_scripts/preprocess_celeba64.py -------------------------------------------------------------------------------- /data_scripts/random_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/data_scripts/random_split.py -------------------------------------------------------------------------------- /data_scripts/utility_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/data_scripts/utility_resize.py -------------------------------------------------------------------------------- /docker_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/docker_run.sh -------------------------------------------------------------------------------- /efficient_vdvae_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_jax/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_jax/data/cifar10_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/data/cifar10_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/data/generic_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/data/generic_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/data/imagenet_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/data/imagenet_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/data/mnist_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/data/mnist_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/hparams.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/hparams.cfg -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/adamax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/adamax.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/autoencoder.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/conv2d.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/div_stats_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/div_stats_utils.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/latent_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/latent_layers.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/layers.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/losses.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/model.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/optimizers.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/schedules.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/model/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/model/ssim.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/reset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/reset.sh -------------------------------------------------------------------------------- /efficient_vdvae_jax/synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/synthesize.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/train.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/denormalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/denormalizer.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/ema_train_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/ema_train_state.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/inference_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/inference_helpers.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/normalizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/normalizer.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/temperature_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/temperature_functions.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/train_helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/train_helpers.py -------------------------------------------------------------------------------- /efficient_vdvae_jax/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_jax/utils/utils.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_torch/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_torch/data/cifar10_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/data/cifar10_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/data/generic_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/data/generic_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/data/imagenet_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/data/imagenet_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/data/mnist_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/data/mnist_data_loader.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/hparams.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/hparams.cfg -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/adamax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/adamax.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/autoencoder.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/conv2d.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/def_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/def_model.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/div_stats_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/div_stats_utils.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/latent_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/latent_layers.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/layers.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/losses.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/model.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/schedules.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/model/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/model/ssim.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/reset.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/reset.sh -------------------------------------------------------------------------------- /efficient_vdvae_torch/synthesize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/synthesize.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/train.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/train.sh -------------------------------------------------------------------------------- /efficient_vdvae_torch/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /efficient_vdvae_torch/utils/temperature_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/utils/temperature_functions.py -------------------------------------------------------------------------------- /efficient_vdvae_torch/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/efficient_vdvae_torch/utils/utils.py -------------------------------------------------------------------------------- /egs/binarized_mnist_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/binarized_mnist_baseline.cfg -------------------------------------------------------------------------------- /egs/celebA64_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebA64_baseline.cfg -------------------------------------------------------------------------------- /egs/celebA64_compact.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebA64_compact.cfg -------------------------------------------------------------------------------- /egs/celebAHQ1024_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebAHQ1024_baseline.cfg -------------------------------------------------------------------------------- /egs/celebAHQ256_5bits_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebAHQ256_5bits_baseline.cfg -------------------------------------------------------------------------------- /egs/celebAHQ256_8bits_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebAHQ256_8bits_baseline.cfg -------------------------------------------------------------------------------- /egs/celebAHQ256_8bits_compact.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/celebAHQ256_8bits_compact.cfg -------------------------------------------------------------------------------- /egs/cifar10_C1.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/cifar10_C1.cfg -------------------------------------------------------------------------------- /egs/cifar10_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/cifar10_baseline.cfg -------------------------------------------------------------------------------- /egs/ffhq1024_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/ffhq1024_baseline.cfg -------------------------------------------------------------------------------- /egs/ffhq256_5bits_C1.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/ffhq256_5bits_C1.cfg -------------------------------------------------------------------------------- /egs/ffhq256_5bits_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/ffhq256_5bits_baseline.cfg -------------------------------------------------------------------------------- /egs/ffhq256_8bits_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/ffhq256_8bits_baseline.cfg -------------------------------------------------------------------------------- /egs/ffhq256_8bits_compact.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/ffhq256_8bits_compact.cfg -------------------------------------------------------------------------------- /egs/imagenet32_C1.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet32_C1.cfg -------------------------------------------------------------------------------- /egs/imagenet32_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet32_baseline.cfg -------------------------------------------------------------------------------- /egs/imagenet32_compact.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet32_compact.cfg -------------------------------------------------------------------------------- /egs/imagenet64_C1.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet64_C1.cfg -------------------------------------------------------------------------------- /egs/imagenet64_baseline.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet64_baseline.cfg -------------------------------------------------------------------------------- /egs/imagenet64_compact.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/egs/imagenet64_compact.cfg -------------------------------------------------------------------------------- /images/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/images/.DS_Store -------------------------------------------------------------------------------- /images/unconditional_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Rayhane-mamah/Efficient-VDVAE/HEAD/images/unconditional_samples.png --------------------------------------------------------------------------------