├── .gitignore ├── LICENSE ├── README.md ├── cmd_utils.py ├── configs ├── cifar10 │ ├── T10.yaml │ ├── T4_ddgan.yaml │ └── cifar10.yaml ├── imagenet64 │ ├── T10.yaml │ ├── T4.yaml │ └── imagenet64.yaml └── lsun │ ├── T10.yaml │ ├── T4.yaml │ ├── T4_wide.yaml │ └── bedroom.yaml ├── evaluations ├── .gitignore ├── README.md ├── evaluator.py └── requirements.txt ├── figure └── DxMI_figure_crop.jpg ├── generate_cifar10.py ├── generate_large.py ├── loader ├── __init__.py └── synthetic.py ├── models ├── DxMI │ ├── __init__.py │ ├── openai_diffusion.py │ ├── trainer.py │ ├── unet_small.py │ └── var_sampler.py ├── __init__.py ├── cm │ ├── __init__.py │ ├── dist_util.py │ ├── dxmi_util.py │ ├── fp16_util.py │ ├── image_datasets.py │ ├── karras_diffusion.py │ ├── logger.py │ ├── nn.py │ ├── random_util.py │ ├── script_util.py │ ├── train_util.py │ └── unet.py ├── diffusion.py ├── logger.py ├── modules.py ├── utils.py └── value.py ├── pytorch_fid ├── fid_score.py └── inception.py ├── requirements.txt ├── train_cifar10.py ├── train_image_large.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/README.md -------------------------------------------------------------------------------- /cmd_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/cmd_utils.py -------------------------------------------------------------------------------- /configs/cifar10/T10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/cifar10/T10.yaml -------------------------------------------------------------------------------- /configs/cifar10/T4_ddgan.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/cifar10/T4_ddgan.yaml -------------------------------------------------------------------------------- /configs/cifar10/cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/cifar10/cifar10.yaml -------------------------------------------------------------------------------- /configs/imagenet64/T10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/imagenet64/T10.yaml -------------------------------------------------------------------------------- /configs/imagenet64/T4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/imagenet64/T4.yaml -------------------------------------------------------------------------------- /configs/imagenet64/imagenet64.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/imagenet64/imagenet64.yaml -------------------------------------------------------------------------------- /configs/lsun/T10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/lsun/T10.yaml -------------------------------------------------------------------------------- /configs/lsun/T4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/lsun/T4.yaml -------------------------------------------------------------------------------- /configs/lsun/T4_wide.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/lsun/T4_wide.yaml -------------------------------------------------------------------------------- /configs/lsun/bedroom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/configs/lsun/bedroom.yaml -------------------------------------------------------------------------------- /evaluations/.gitignore: -------------------------------------------------------------------------------- 1 | classify_image_graph_def.pb 2 | -------------------------------------------------------------------------------- /evaluations/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/evaluations/README.md -------------------------------------------------------------------------------- /evaluations/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/evaluations/evaluator.py -------------------------------------------------------------------------------- /evaluations/requirements.txt: -------------------------------------------------------------------------------- 1 | tensorflow>=2.0 2 | scipy 3 | requests 4 | tqdm 5 | -------------------------------------------------------------------------------- /figure/DxMI_figure_crop.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/figure/DxMI_figure_crop.jpg -------------------------------------------------------------------------------- /generate_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/generate_cifar10.py -------------------------------------------------------------------------------- /generate_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/generate_large.py -------------------------------------------------------------------------------- /loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/loader/__init__.py -------------------------------------------------------------------------------- /loader/synthetic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/loader/synthetic.py -------------------------------------------------------------------------------- /models/DxMI/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/DxMI/openai_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/DxMI/openai_diffusion.py -------------------------------------------------------------------------------- /models/DxMI/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/DxMI/trainer.py -------------------------------------------------------------------------------- /models/DxMI/unet_small.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/DxMI/unet_small.py -------------------------------------------------------------------------------- /models/DxMI/var_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/DxMI/var_sampler.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/cm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/cm/dist_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/dist_util.py -------------------------------------------------------------------------------- /models/cm/dxmi_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/dxmi_util.py -------------------------------------------------------------------------------- /models/cm/fp16_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/fp16_util.py -------------------------------------------------------------------------------- /models/cm/image_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/image_datasets.py -------------------------------------------------------------------------------- /models/cm/karras_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/karras_diffusion.py -------------------------------------------------------------------------------- /models/cm/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/logger.py -------------------------------------------------------------------------------- /models/cm/nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/nn.py -------------------------------------------------------------------------------- /models/cm/random_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/random_util.py -------------------------------------------------------------------------------- /models/cm/script_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/script_util.py -------------------------------------------------------------------------------- /models/cm/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/train_util.py -------------------------------------------------------------------------------- /models/cm/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/cm/unet.py -------------------------------------------------------------------------------- /models/diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/diffusion.py -------------------------------------------------------------------------------- /models/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/logger.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/modules.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/value.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/models/value.py -------------------------------------------------------------------------------- /pytorch_fid/fid_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/pytorch_fid/fid_score.py -------------------------------------------------------------------------------- /pytorch_fid/inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/pytorch_fid/inception.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | omegaconf 2 | hydra-core 3 | tqdm 4 | scikit-learn 5 | -------------------------------------------------------------------------------- /train_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/train_cifar10.py -------------------------------------------------------------------------------- /train_image_large.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/train_image_large.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/swyoon/Diffusion-by-MaxEntIRL/HEAD/utils.py --------------------------------------------------------------------------------