/ODEOT

use neural ODE to train for an optimal transport

Primary LanguagePython

#ODEOT

Dependencies

fml torchdiffeq open3d

pay attention that open3d has to be imported before torch