diff deep/stacked_dae/v_sylvain/nist_sda.py @ 235:ecb69e17950b

correction de bugs
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Sun, 14 Mar 2010 20:25:12 -0400
parents c452e3a0a3b1
children 9fc641d7adda
line wrap: on
line diff
--- a/deep/stacked_dae/v_sylvain/nist_sda.py	Sun Mar 14 15:17:04 2010 -0400
+++ b/deep/stacked_dae/v_sylvain/nist_sda.py	Sun Mar 14 20:25:12 2010 -0400
@@ -120,10 +120,10 @@
                                     n_ins=n_ins, n_outs=n_outs,\
                                     series=series)
 
-    optimizer.pretrain()
+    optimizer.pretrain(datasets.nist_all)
     channel.save()
 
-    optimizer.finetune()
+    optimizer.finetune(datasets.nist_all)
     channel.save()
 
     return channel.COMPLETE