Mercurial > ift6266

diff deep/stacked_dae/v_sylvain/nist_sda.py @ 307:ed0443f7aad4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Ajout d'une protection supplementaire (un peu de paranoia) ainsi que commentaires
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Wed, 31 Mar 2010 21:03:41 -0400
parents 698313f8f6e6
children bd6085d77706
line wrap: on
line diff
--- a/deep/stacked_dae/v_sylvain/nist_sda.py	Wed Mar 31 21:02:27 2010 -0400
+++ b/deep/stacked_dae/v_sylvain/nist_sda.py	Wed Mar 31 21:03:41 2010 -0400
@@ -75,7 +75,7 @@
         optimizer.pretrain(datasets.nist_all()) 
     elif state['pretrain_choice'] == 1:
         #To know how many file will be used during pretraining
-        nb_file = state['pretraining_epochs_per_layer'] 
+        nb_file = int(state['pretraining_epochs_per_layer']) 
         state['pretraining_epochs_per_layer'] = 1 #Only 1 time over the dataset
         if nb_file >=100:
             sys.exit("The code does not support this much pretraining epoch (99 max with P07).\n"+