changeset 307:ed0443f7aad4

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 a78dbbc61f37
children a76bae0f2388
files deep/stacked_dae/v_sylvain/nist_sda.py
diffstat 1 files changed, 1 insertions(+), 1 deletions(-) [+]
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"+