diff deep/stacked_dae/v_sylvain/nist_sda.py @ 390:7c201ca1484f

Correction d'un bug mineurpour le nom de la serie h5
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Tue, 27 Apr 2010 08:55:16 -0400
parents 8875853b5bfc
children c91d7b67fa41
line wrap: on
line diff
--- a/deep/stacked_dae/v_sylvain/nist_sda.py	Tue Apr 27 08:43:04 2010 -0400
+++ b/deep/stacked_dae/v_sylvain/nist_sda.py	Tue Apr 27 08:55:16 2010 -0400
@@ -74,6 +74,8 @@
         nom_serie="series_NIST.h5"
     elif state['pretrain_choice'] == 1:
         nom_serie="series_P07.h5"
+    elif state['pretrain_choice'] == 2:
+        nom_serie="series_PNIST.h5"
 
     series = create_series(state.num_hidden_layers,nom_serie)