Skip to content

Code for our TMLR '24 Journal: MMD-Regularized UOT.

License

Notifications You must be signed in to change notification settings

Piyushi-0/MMD-reg-OT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

44 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

  • 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 run pip install .

Algorithms

Basic Examples

Video Presentation

If you find this useful, consider giving ato this repository & citing our work.