changeset 1508:b28e8730c948

fix test.
author Frederic Bastien <nouiz@nouiz.org>
date Mon, 12 Sep 2011 11:45:56 -0400
parents 2a6a6f16416c
children b709f6b53b17
files pylearn/algorithms/tests/test_mcRBM.py
diffstat 1 files changed, 2 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/pylearn/algorithms/tests/test_mcRBM.py	Mon Sep 12 11:45:41 2011 -0400
+++ b/pylearn/algorithms/tests/test_mcRBM.py	Mon Sep 12 11:45:56 2011 -0400
@@ -15,7 +15,8 @@
 def _default_rbm_alloc(n_I, n_K=256, n_J=100):
     return mcRBM.alloc(n_I, n_K, n_J)
 
-def _default_trainer_alloc(rbm, train_batch, batchsize, l1_penalty, l1_penalty_start):
+def _default_trainer_alloc(rbm, train_batch, batchsize, initial_lr_per_example,
+                           l1_penalty, l1_penalty_start, persistent_chains):
     return mcRBMTrainer.alloc(rbm, train_batch, batchsize, l1_penalty=l1_penalty,
             l1_penalty_start=l1_penalty_start,persistent_chains=persistent_chains)