diff options
| author | AdrienCorenflos <adrien.corenflos@gmail.com> | 2021-04-19 14:57:51 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-04-19 13:57:51 +0200 |
| commit | 2a3f2241951ea9cc044b4fba8a382b6ae9630513 (patch) | |
| tree | c4a07fda0e2ac6495d673df8aba277588bb47783 /ot/gpu | |
| parent | 3a2ec71ae7d11aa650a7d3222357885010a9b2c3 (diff) | |
BUG/DOC FIX - Sinkhorn divergence used the wrong weights, and sinkhorn2 didn't support epsilon_scaling method. (#235)
* FIX:
1. Documentation of loss specific functions
2. Sinkhorn divergence weights handling
3. Sinkhorn2 does not support epsilon scaling, so I removed it (it *should* arguably support it, but this would require a refactoring of the sinkhorn iterates pretty much everywhere, maybe should be done in torch first?)
* Had some PEP8 issues
Co-authored-by: Rémi Flamary <remi.flamary@gmail.com>
Diffstat (limited to 'ot/gpu')
0 files changed, 0 insertions, 0 deletions