vivienseguy / Large-Scale-OT

Stochastic algorithms for computing Regularized Optimal Transport
55 stars 11 forks source link

Large-Scale-OT with GPU #4

Open tuln128 opened 2 years ago

tuln128 commented 2 years ago

Hi Vivien, Thanks for such a nice framework. I am planning to employ your method on our data. However, I am wondering how to make Large-Scale-OT run on GPU. In fact, I have tried to set device_type="gpu" but it caused runtime error (probably some modules have not been implemented to work on GPU settings). Could you please let me know how to solve this problem (or correct me if I am wrong). I am looking forward to hearing your response. Kind regards,