comparison deep/convolutional_dae/run_exp.py @ 376:01445a75c702

Fix filter sizes to have meaningful networks with more that 2 layers.
author Arnaud Bergeron <abergeron@gmail.com>
date Sun, 25 Apr 2010 17:10:09 -0400
parents 69109e41983e
children
comparison
equal deleted inserted replaced
335:5ddb1878dfbc 376:01445a75c702
20 channel.save() 20 channel.save()
21 21
22 dset = datasets.nist_P07() 22 dset = datasets.nist_P07()
23 23
24 nfilts = [] 24 nfilts = []
25 fsizes = []
25 if state.nfilts1 != 0: 26 if state.nfilts1 != 0:
26 nfilts.append(state.nfilts1) 27 nfilts.append(state.nfilts1)
28 fsizes.append((5,5))
27 if state.nfilts2 != 0: 29 if state.nfilts2 != 0:
28 nfilts.append(state.nfilts2) 30 nfilts.append(state.nfilts2)
31 fsizes.append((3,3))
29 if state.nfilts3 != 0: 32 if state.nfilts3 != 0:
30 nfilts.append(state.nfilts3) 33 nfilts.append(state.nfilts3)
34 fsizes.append((3,3))
31 if state.nfilts4 != 0: 35 if state.nfilts4 != 0:
32 nfilts.append(state.nfilts4) 36 nfilts.append(state.nfilts4)
37 fsizes.append((2,2))
33 38
34 fsizes = [(5,5)]*len(nfilts)
35 subs = [(2,2)]*len(nfilts) 39 subs = [(2,2)]*len(nfilts)
36 noise = [state.noise]*len(nfilts) 40 noise = [state.noise]*len(nfilts)
37 41
38 pretrain_funcs, trainf, evalf, net = build_funcs( 42 pretrain_funcs, trainf, evalf, net = build_funcs(
39 img_size=(32, 32), 43 img_size=(32, 32),