cnn : reduce patch size

tao-pr · Apr 17, 2017 · 3112178 · 3112178
1 parent a6c9a35
commit 3112178
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/model/cnn.py b/model/cnn.py
@@ -46,7 +46,7 @@ def new(self, image_dim, final_vec_dim):
       l_conv1 = layers.Conv2DLayer(l_max0, 32, (5,5))
       l_max1  = layers.MaxPool2DLayer(l_conv1, (5,5), stride=2)
       l_conv2 = layers.Conv2DLayer(l_conv1, 32, (3,3))
-      l_pool  = layers.MaxPool2DLayer(l_conv2, (5,5), stride=1)
+      l_pool  = layers.MaxPool2DLayer(l_conv2, (3,3), stride=1)
       l_1d1   = layers.DenseLayer(l_pool, 24)
       l_1d2   = layers.DenseLayer(l_1d1, 8)
       l_1d3   = layers.DenseLayer(l_1d2, 1)