Large-Scale-OT icon indicating copy to clipboard operation
Large-Scale-OT copied to clipboard

Stochastic algorithms for computing Regularized Optimal Transport

Results 2 Large-Scale-OT issues
Sort by recently updated
recently updated
newest added

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...

when I run simple.py "reg = - alpha * torch.exp(((u[:, None] + v[None, :] - K) / alpha))“ it seems that reg=nan. why does this happen?