.
├── bin
│   ├── cifar10.sh
│   └── mnist.sh
├── data
├── figs
│   └── gif
├── models
│   ├── __init__.py
│   ├── cifar10.py
│   ├── gaussian_model.py
│   └── mnist.py
├── src
│   ├── __init__.py
│   ├── main.py
│   ├── qpwgan.py
│   ├── discrete_measures.py
│   ├── gaussian_mixture.py
│   ├── metrics.py
│   ├── plot_nearest_distance.py
│   └── utils.py
├── README.md
├── requirements.txt
└── setup.py

pip install -r requirements.txt
pip install -e .

wandb login

python src/discrete_measures.py

python src/gaussian_mixture.py \
                 --n_epoch 601 \
                 --search_space full \
                 --n_critic_iter 2 \
                 --reg_coef1 0.1 \
                 --reg_coef2 1 \
                 --batch_size 64

bash bin/mnist.sh

bash bin/cifar10.sh