Skip to content

Commit 31aa5ac

Browse files
committed
update doc gpu fixes
1 parent 4efdda7 commit 31aa5ac

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

ot/gpu/da.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -181,15 +181,6 @@ def sinkhorn_lpl1_mm(a, labels_a, b, M_GPU, reg, eta=0.1, numItermax=10,
181181

182182
class OTDA_GPU(OTDA):
183183
def normalizeM(self, norm):
184-
""" Apply normalization to the loss matrix
185-
186-
187-
Parameters
188-
----------
189-
norm : str
190-
type of normalization from 'median','max','log','loglog'
191-
192-
"""
193184
if norm == "median":
194185
self.M_GPU.divide(float(np.median(self.M_GPU.asarray())))
195186
elif norm == "max":

0 commit comments

Comments
 (0)