This repository contains two projects:
- Sinkhorn Barycenter via Functional Gradient Descent
- Sinkhorn Natural Gradient
To run SD on MNIST, please first run MNIST_to_measure.py for data preprocessing.
geomloss
torch
tensorboard
The following baselines are included:
- Sinkhorn GAN
- Gradual GAN
- MMD GAN
- OT GAN
- Wasserstein GAN
geomloss
torch
torchvision
tensorboard