changeset 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 5ddb1878dfbc
children 0b7e64e8e93f
files deep/convolutional_dae/run_exp.py
diffstat 1 files changed, 5 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/deep/convolutional_dae/run_exp.py	Thu Apr 15 12:53:03 2010 -0400
+++ b/deep/convolutional_dae/run_exp.py	Sun Apr 25 17:10:09 2010 -0400
@@ -22,16 +22,20 @@
     dset = datasets.nist_P07()
 
     nfilts = []
+    fsizes = []
     if state.nfilts1 != 0:
         nfilts.append(state.nfilts1)
+        fsizes.append((5,5))
         if state.nfilts2 != 0:
             nfilts.append(state.nfilts2)
+            fsizes.append((3,3))
             if state.nfilts3 != 0:
                 nfilts.append(state.nfilts3)
+                fsizes.append((3,3))
                 if state.nfilts4 != 0:
                     nfilts.append(state.nfilts4)
+                    fsizes.append((2,2))
 
-    fsizes = [(5,5)]*len(nfilts)
     subs = [(2,2)]*len(nfilts)
     noise = [state.noise]*len(nfilts)