├── LICENSE ├── README.md ├── compute.py ├── configs ├── sampling.yaml └── training.yaml ├── helpers.py ├── langevin_cifar.gif ├── langevin_imagenet.gif ├── lmc_cifar.gif ├── lmc_imagenet.gif ├── main.py ├── metrics ├── LICENSE ├── README.md ├── __init__.py ├── fid_pytorch.py ├── inception.py └── res │ └── stats_pytorch │ └── README.md ├── models ├── discriminator.py ├── energy_model.py ├── generator.py ├── made_model.py ├── mog_maf_model.py └── toy_models.py ├── requirements.txt ├── samplers.py ├── scripts.sh ├── trainer.py └── utils ├── celebA.py ├── dataloader.py ├── draw.py ├── fid_scheduler.py ├── imagenet.py ├── kid_score.py ├── timer.py └── vizualization.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/README.md -------------------------------------------------------------------------------- /compute.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/compute.py -------------------------------------------------------------------------------- /configs/sampling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/configs/sampling.yaml -------------------------------------------------------------------------------- /configs/training.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/configs/training.yaml -------------------------------------------------------------------------------- /helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/helpers.py -------------------------------------------------------------------------------- /langevin_cifar.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/langevin_cifar.gif -------------------------------------------------------------------------------- /langevin_imagenet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/langevin_imagenet.gif -------------------------------------------------------------------------------- /lmc_cifar.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/lmc_cifar.gif -------------------------------------------------------------------------------- /lmc_imagenet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/lmc_imagenet.gif -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/main.py -------------------------------------------------------------------------------- /metrics/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/metrics/LICENSE -------------------------------------------------------------------------------- /metrics/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/metrics/README.md -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/fid_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/metrics/fid_pytorch.py -------------------------------------------------------------------------------- /metrics/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/metrics/inception.py -------------------------------------------------------------------------------- /metrics/res/stats_pytorch/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/metrics/res/stats_pytorch/README.md -------------------------------------------------------------------------------- /models/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/discriminator.py -------------------------------------------------------------------------------- /models/energy_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/energy_model.py -------------------------------------------------------------------------------- /models/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/generator.py -------------------------------------------------------------------------------- /models/made_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/made_model.py -------------------------------------------------------------------------------- /models/mog_maf_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/mog_maf_model.py -------------------------------------------------------------------------------- /models/toy_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/models/toy_models.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/requirements.txt -------------------------------------------------------------------------------- /samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/samplers.py -------------------------------------------------------------------------------- /scripts.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/scripts.sh -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/trainer.py -------------------------------------------------------------------------------- /utils/celebA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/celebA.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /utils/draw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/draw.py -------------------------------------------------------------------------------- /utils/fid_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/fid_scheduler.py -------------------------------------------------------------------------------- /utils/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/imagenet.py -------------------------------------------------------------------------------- /utils/kid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/kid_score.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/vizualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/MichaelArbel/GeneralizedEBM/HEAD/utils/vizualization.py --------------------------------------------------------------------------------