Mercurial > ift6266
diff deep/convolutional_dae/run_exp.py @ 293:d89820070ea0
Add some prints to see the current step.
author | Arnaud Bergeron <abergeron@gmail.com> |
---|---|
date | Fri, 26 Mar 2010 19:18:03 -0400 |
parents | 8108d271c30c |
children | 8babd43235dd |
line wrap: on
line diff
--- a/deep/convolutional_dae/run_exp.py Fri Mar 26 18:49:27 2010 -0400 +++ b/deep/convolutional_dae/run_exp.py Fri Mar 26 19:18:03 2010 -0400 @@ -9,6 +9,7 @@ from ift6266.deep.convolutional_dae.sgd_opt import sgd_opt import pylearn, theano, ift6266 import pylearn.version + import sys # params: bsize, pretrain_lr, train_lr, nfilts1, nfilts2, nftils3, nfilts4 # pretrain_rounds, noise, mlp_sz @@ -17,7 +18,7 @@ # TODO: maybe record pynnet version? channel.save() - dset = datasets.nist_all(1000) + dset = datasets.nist_digits() nfilts = [] if state.nfilts1 != 0: @@ -53,11 +54,15 @@ series = create_series() + print "pretraining ..." + sys.stdout.flush() do_pretrain(pretrain_fs, state.pretrain_rounds, series['recons_error']) + print "training ..." + sys.stdout.flush() sgd_opt(train, valid, test, training_epochs=100000, patience=10000, patience_increase=2., improvement_threshold=0.995, - validation_frequency=2500, series=series, net=net) + validation_frequency=1000, series=series, net=net) if __name__ == '__main__': st = dumb()