Fix bug in multiloss decoder

MarvinTeichmann · Mar 20, 2017 · 2784122 · 2784122
1 parent 2fd7d9e
commit 2784122
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/encoder/fcn8_vgg.py b/encoder/fcn8_vgg.py
@@ -53,6 +53,6 @@ def inference(hypes, images, train=True):
 
     logits['feed4'] = vgg_fcn.pool3
 
-    # logits['fcn_logits'] = vgg_fcn.upscore32
+    logits['fcn_logits'] = vgg_fcn.upscore32
 
     return logits