diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2021-11-04 15:19:57 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-11-04 15:19:57 +0100 |
commit | 0e431c203a66c6d48e6bb1efeda149460472a0f0 (patch) | |
tree | 22a447a1dbb1505b18f9e426e1761cf6b328b6eb /ot/sliced.py | |
parent | 2fe69eb130827560ada704bc25998397c4357821 (diff) |
[MRG] Add tests about type and GPU for emd/emd2 + 1d variants + wasserstein1d (#304)
* new test gpu
* pep 8 of couse
* debug torch
* jax with gpu
* device put
* device put
* it works
* emd1d and emd2_1d working
* emd_1d and emd2_1d done
* cleanup
* of course
* should work on gpu now
* tests done+ pep8
Diffstat (limited to 'ot/sliced.py')
0 files changed, 0 insertions, 0 deletions