Skip to content

Commit 49af122

Browse files
committed
correction bug merge ot.gpu.da
1 parent 287c4c0 commit 49af122

File tree

1 file changed

+0
-18
lines changed

1 file changed

+0
-18
lines changed

ot/gpu/da.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -134,23 +134,6 @@ def sinkhorn_lpl1_mm(a, labels_a, b, M, reg, eta=0.1, numItermax=10,
134134
W = np.ones(M.shape)
135135
for (i, c) in enumerate(classes):
136136

137-
138-
<< << << < HEAD
139-
(_, nbRow) = indices_labels[i].shape
140-
tmpC_GPU = cudamat.empty((Nfin, nbRow)).assign(0)
141-
transp_GPU.transpose().select_columns(indices_labels[i], tmpC_GPU)
142-
majs_GPU = tmpC_GPU.sum(axis=1).add(epsilon)
143-
cudamat.pow(majs_GPU, (p - 1))
144-
majs_GPU.mult(p)
145-
146-
tmpC_GPU.assign(0)
147-
tmpC_GPU.add_col_vec(majs_GPU)
148-
W_GPU.set_selected_columns(indices_labels[i], tmpC_GPU)
149-
150-
W_GPU = W_GPU.transpose()
151-
152-
return transp_GPU.asarray()
153-
=======
154137
majs = np.sum(transp[indices_labels[i]], axis=0)
155138
majs = p * ((majs + epsilon)**(p - 1))
156139
W[indices_labels[i]] = majs
@@ -159,4 +142,3 @@ def sinkhorn_lpl1_mm(a, labels_a, b, M, reg, eta=0.1, numItermax=10,
159142
return utils.to_np(transp)
160143
else:
161144
return transp
162-
>>>>>>> master

0 commit comments

Comments
 (0)