diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 15:45:09 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2017-07-24 15:45:09 +0200 |
| commit | bd705ed847dd7e43082e9d2771a59e539d6b7440 (patch) | |
| tree | 6aa66499a61a487eebe59b3c75cdbec97e7019b1 /test/test_gpu.py | |
| parent | 33f3d309209baa8c5e127d02f00aae0660ed7bfb (diff) | |
add test yunmlix and bary
Diffstat (limited to 'test/test_gpu.py')
| -rw-r--r-- | test/test_gpu.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/test_gpu.py b/test/test_gpu.py index 9cc39d7..24797f2 100644 --- a/test/test_gpu.py +++ b/test/test_gpu.py @@ -48,7 +48,7 @@ def test_gpu_sinkhorn_lpl1(): print("min:{:.3E}, max:{:.3E}, mean:{:.3E}, std:{:.3E}" .format(np.min(r), np.max(r), np.mean(r), np.std(r))) - for n in [50, 100, 500, 1000]: + for n in [50, 100, 500]: print(n) a = np.random.rand(n // 4, 100) labels_a = np.random.randint(10, size=(n // 4)) |