Merge pull request alexjc#45 from jwinder1/bugfix_sem2_1

Use conv2_1 layer instead of conv3_1 for sem2_1 input
dupuleng · Mar 13, 2016 · deeff95 · deeff95
2 parents a660f1f + 9c97bce
commit deeff95
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']])