Skip to content

Commit ccbe274

Browse files
committed
speedup test stochastic
1 parent 7786e85 commit ccbe274

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/test_stochastic.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def test_stochastic_sag():
3232
# test sag
3333
n = 15
3434
reg = 1
35-
numItermax = 300000
35+
numItermax = 30000
3636
rng = np.random.RandomState(0)
3737

3838
x = rng.randn(n, 2)
@@ -62,7 +62,7 @@ def test_stochastic_asgd():
6262
# test asgd
6363
n = 15
6464
reg = 1
65-
numItermax = 300000
65+
numItermax = 100000
6666
rng = np.random.RandomState(0)
6767

6868
x = rng.randn(n, 2)
@@ -92,7 +92,7 @@ def test_sag_asgd_sinkhorn():
9292
# test all algorithms
9393
n = 15
9494
reg = 1
95-
nb_iter = 30000
95+
nb_iter = 100000
9696
rng = np.random.RandomState(0)
9797

9898
x = rng.randn(n, 2)

0 commit comments

Comments
 (0)