The code is attached to the paper ``An Accelerated Stochastic Algorithm for Solving the Optimal Transport Problem''(https://arxiv.org/abs/2203.00813), Yiling Xie, Yiling Luo, Xiaoming Huo.

Download the file and run CompareMNIST.py to see the comparison on MNIST dataset, Comparesyn.py to see the comparison on synthetic data.