Skip to content

Commit b331460

Browse files
committed
small normalization bug
1 parent ad693e3 commit b331460

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

README.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,8 @@ cd cudamat
6868
python setup.py install --user # for user install (no root)
6969
```
7070

71+
obviously you need CUDA installed and a compatible GPU.
72+
7173
## Examples
7274

7375
The examples folder contain several examples and use case for the library. The full documentation is available on [Readthedocs](http://pot.readthedocs.io/)

ot/da.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -620,7 +620,7 @@ def fit(self,xs,xt,ws=None,wt=None,norm=None):
620620
self.wt=wt
621621

622622
self.M=dist(xs,xt,metric=self.metric)
623-
self.normalize()
623+
self.normalizeM(norm)
624624
self.G=emd(ws,wt,self.M)
625625
self.computed=True
626626

0 commit comments

Comments
 (0)