Use conv2_1 layer instead of conv3_1 for sem2_1 input

521314 · Mar 13, 2016 · 9c97bce · 9c97bce
1 parent a660f1f
commit 9c97bce
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doodle.py b/doodle.py
@@ -128,7 +128,7 @@ def setup_model(self):
         net['map_3'] = PoolLayer(net['map'], 4, mode='average_exc_pad')
         net['map_4'] = PoolLayer(net['map'], 8, mode='average_exc_pad')
 
-        net['sem2_1'] = ConcatLayer([net['conv3_1'], net['map_2']])
+        net['sem2_1'] = ConcatLayer([net['conv2_1'], net['map_2']])
         net['sem3_1'] = ConcatLayer([net['conv3_1'], net['map_3']])
         net['sem4_1'] = ConcatLayer([net['conv4_1'], net['map_4']])