Mercurial > ift6266
comparison deep/stacked_dae/v_sylvain/nist_sda_retrieve.py @ 311:8b31280129a9
Petits changements pour pouvoir utiliser le GPU
author | SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca> |
---|---|
date | Thu, 01 Apr 2010 13:44:30 -0400 |
parents | a76bae0f2388 |
children | e301a2f32665 |
comparison
equal
deleted
inserted
replaced
310:a5abd5c8b4b0 | 311:8b31280129a9 |
---|---|
63 optimizer = SdaSgdOptimizer(dataset=datasets.nist_all(), | 63 optimizer = SdaSgdOptimizer(dataset=datasets.nist_all(), |
64 hyperparameters=state, \ | 64 hyperparameters=state, \ |
65 n_ins=n_ins, n_outs=n_outs,\ | 65 n_ins=n_ins, n_outs=n_outs,\ |
66 examples_per_epoch=examples_per_epoch, \ | 66 examples_per_epoch=examples_per_epoch, \ |
67 series=series, | 67 series=series, |
68 max_minibatches=rtt) | 68 max_minibatches=50) #To correct later ! |
69 | 69 |
70 parameters=[] | 70 parameters=[] |
71 #Number of files of P07 used for pretraining | 71 #Number of files of P07 used for pretraining |
72 nb_file=0 | 72 nb_file=0 |
73 ## if state['pretrain_choice'] == 0: | 73 ## if state['pretrain_choice'] == 0: |