diff deep/stacked_dae/v_sylvain/nist_sda_retrieve.py @ 387:44094ff28ed1

Ajout d'une option finetune amelioree pour PNIST07
author SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca>
date Tue, 27 Apr 2010 08:41:58 -0400
parents 5c3935aa3f8a
children d76c85ba12d6
line wrap: on
line diff
--- a/deep/stacked_dae/v_sylvain/nist_sda_retrieve.py	Tue Apr 27 08:41:44 2010 -0400
+++ b/deep/stacked_dae/v_sylvain/nist_sda_retrieve.py	Tue Apr 27 08:41:58 2010 -0400
@@ -151,7 +151,7 @@
         print ('\n\n\tFinetune with PNIST07\n\n')
         optimizer.reload_parameters(PATH+'params_pretrain.txt')
         optimizer.finetune(datasets.PNIST07(),datasets.nist_all(),max_finetune_epoch_NIST,ind_test=2,decrease=decrease_lr,dataset_test2=datasets.nist_P07())    
-            
+        optimizer.finetune(datasets.nist_all(),datasets.PNIST07(),max_finetune_epoch_NIST,ind_test=31,decrease=decrease_lr,dataset_test2=datasets.nist_P07())    
         
     if finetune_choice==-1:
         print('\nSERIE OF 4 DIFFERENT FINETUNINGS')
@@ -180,6 +180,9 @@
         sys.stdout.flush()
         optimizer.reload_parameters(PATH+'params_pretrain.txt')
         optimizer.finetune(datasets.PNIST07(),datasets.nist_all(),max_finetune_epoch_NIST,ind_test=2,decrease=decrease_lr,dataset_test2=datasets.nist_P07())    
+        channel.save()
+        sys.stdout.flush()
+        optimizer.finetune(datasets.nist_all(),datasets.PNIST07(),max_finetune_epoch_NIST,ind_test=31,decrease=decrease_lr,dataset_test2=datasets.nist_P07())    
          
     
     channel.save()