Code for our TMLR Journal: MMD-Regularized Unbalanced Optimal Transport
- You may first install PyTorch and Torchvision via
conda install pytorch==1.13.1 torchvision==0.14.1 pytorch-cuda=11.6 -c pytorch -c nvidia
. - The other packages used in this repository can be installed via
pip install -r requirements.txt
. - To install our
ot_mmd
package, please clone this repository and runpip install .
- Code for solving MMD-OT using Accelerated PGD.
- Code for solving the MMD-OT barycenter problem using Accelerated PGD.
If you find this useful, consider giving a โญ to this repository & citing our work.