fix debug display condition (#432)

blue-oil · Sep 18, 2019 · 314355b · 314355b
1 parent 34758e3
commit 314355b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lmnet/lmnet/networks/segmentation/base.py b/lmnet/lmnet/networks/segmentation/base.py
@@ -178,7 +178,8 @@ def loss(self, output, labels):
                 weight = (all_size - num_label) / all_size
                 loss_weight.append(weight)
 
-            loss_weight = tf.Print(loss_weight, loss_weight, message="loss_weight:")
+            if self.is_debug:
+                loss_weight = tf.Print(loss_weight, loss_weight, message="loss_weight:")
 
             reshape_output = tf.reshape(output, (-1, self.num_classes))