diff deep/stacked_dae/nist_sda.py @ 268:b077d9e97a3b

Légères corrections à config.py.example et ajusté nist_sda pour utiliser la nouvelle 'interface' des datasets (nist_all()).
author fsavard
date Fri, 19 Mar 2010 11:31:57 -0400
parents 798d1344e6a2
children 7b4507295eba 43afd29f3dbd
line wrap: on
line diff
--- a/deep/stacked_dae/nist_sda.py	Fri Mar 19 11:12:40 2010 -0400
+++ b/deep/stacked_dae/nist_sda.py	Fri Mar 19 11:31:57 2010 -0400
@@ -61,17 +61,17 @@
 
     print "Creating optimizer with state, ", state
 
-    optimizer = SdaSgdOptimizer(dataset=datasets.nist_all, 
+    optimizer = SdaSgdOptimizer(dataset=datasets.nist_all(), 
                                     hyperparameters=state, \
                                     n_ins=n_ins, n_outs=n_outs,\
                                     examples_per_epoch=examples_per_epoch, \
                                     series=series,
                                     max_minibatches=rtt)
 
-    optimizer.pretrain(datasets.nist_all)
+    optimizer.pretrain(datasets.nist_all())
     channel.save()
 
-    optimizer.finetune(datasets.nist_all)
+    optimizer.finetune(datasets.nist_all())
     channel.save()
 
     return channel.COMPLETE