Skip to content

Commit 884e080

Browse files
committed
Lower the test time to make it pass on travis
1 parent 4449866 commit 884e080

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

code/test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def test_dbn():
4747

4848
def test_rbm():
4949
rbm.test_rbm(training_epochs=1, batch_size=300, n_chains=1, n_samples=1,
50-
output_folder='tmp_rbm_plots')
50+
n_hidden=100, output_folder='tmp_rbm_plots')
5151

5252

5353
def speed():

0 commit comments

Comments
 (0)