├── .gitignore ├── LICENCE ├── README.md ├── assets ├── diagram.png └── samples.png ├── datasets.yml ├── experiments ├── calc_FID.py ├── calc_PRDC.py ├── calc_VQGAN_FID.py ├── calc_approximate_ELBO.py ├── calc_nearest_neighbours.py ├── generate_big_samples.py └── get_samples.py ├── hparams ├── __init__.py ├── defaults │ ├── __init__.py │ ├── base.py │ ├── experiment_defaults.py │ ├── sampler_defaults.py │ └── vqgan_defaults.py └── set_up_hparams.py ├── models ├── __init__.py ├── absorbing_diffusion.py ├── autoregressive.py ├── diffaug.py ├── helpers.py ├── sampler.py ├── transformer.py └── vqgan.py ├── requirements.yml ├── train_sampler.py ├── train_vqgan.py └── utils ├── __init__.py ├── data_utils.py ├── experiment_utils.py ├── log_utils.py ├── sampler_utils.py ├── train_utils.py └── vqgan_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENCE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/LICENCE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/README.md -------------------------------------------------------------------------------- /assets/diagram.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/assets/diagram.png -------------------------------------------------------------------------------- /assets/samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/assets/samples.png -------------------------------------------------------------------------------- /datasets.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/datasets.yml -------------------------------------------------------------------------------- /experiments/calc_FID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/calc_FID.py -------------------------------------------------------------------------------- /experiments/calc_PRDC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/calc_PRDC.py -------------------------------------------------------------------------------- /experiments/calc_VQGAN_FID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/calc_VQGAN_FID.py -------------------------------------------------------------------------------- /experiments/calc_approximate_ELBO.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/calc_approximate_ELBO.py -------------------------------------------------------------------------------- /experiments/calc_nearest_neighbours.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/calc_nearest_neighbours.py -------------------------------------------------------------------------------- /experiments/generate_big_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/generate_big_samples.py -------------------------------------------------------------------------------- /experiments/get_samples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/experiments/get_samples.py -------------------------------------------------------------------------------- /hparams/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/__init__.py -------------------------------------------------------------------------------- /hparams/defaults/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /hparams/defaults/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/defaults/base.py -------------------------------------------------------------------------------- /hparams/defaults/experiment_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/defaults/experiment_defaults.py -------------------------------------------------------------------------------- /hparams/defaults/sampler_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/defaults/sampler_defaults.py -------------------------------------------------------------------------------- /hparams/defaults/vqgan_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/defaults/vqgan_defaults.py -------------------------------------------------------------------------------- /hparams/set_up_hparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/hparams/set_up_hparams.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/absorbing_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/absorbing_diffusion.py -------------------------------------------------------------------------------- /models/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/autoregressive.py -------------------------------------------------------------------------------- /models/diffaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/diffaug.py -------------------------------------------------------------------------------- /models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/helpers.py -------------------------------------------------------------------------------- /models/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/sampler.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/models/vqgan.py -------------------------------------------------------------------------------- /requirements.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/requirements.yml -------------------------------------------------------------------------------- /train_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/train_sampler.py -------------------------------------------------------------------------------- /train_vqgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/train_vqgan.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/experiment_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/experiment_utils.py -------------------------------------------------------------------------------- /utils/log_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/log_utils.py -------------------------------------------------------------------------------- /utils/sampler_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/sampler_utils.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/vqgan_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/samb-t/unleashing-transformers/HEAD/utils/vqgan_utils.py --------------------------------------------------------------------------------