diff options
author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-14 08:58:56 +0200 |
---|---|---|
committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-14 08:58:56 +0200 |
commit | 7786e8502ea010a52918861014ab0e7a934a7d45 (patch) | |
tree | 42c79b0503cb19f69a8520207e82869ae9d72915 /test | |
parent | 05eec20390f453c3ba57ff68fd72a30375f78234 (diff) |
correct path function in example
Diffstat (limited to 'test')
-rw-r--r-- | test/test_stochastic.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/test/test_stochastic.py b/test/test_stochastic.py index 0128317..62cf82a 100644 --- a/test/test_stochastic.py +++ b/test/test_stochastic.py @@ -92,7 +92,7 @@ def test_sag_asgd_sinkhorn(): # test all algorithms n = 15 reg = 1 - nb_iter = 300000 + nb_iter = 30000 rng = np.random.RandomState(0) x = rng.randn(n, 2) @@ -167,7 +167,7 @@ def test_dual_sgd_sinkhorn(): # test all dual algorithms n = 10 reg = 1 - nb_iter = 150000 + nb_iter = 15000 batch_size = 10 rng = np.random.RandomState(0) |