test loss function

Arno1235 · Mar 20, 2023 · 82358c4 · 82358c4
1 parent 5a031ad
commit 82358c4
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/unet_2d.py b/unet_2d.py
@@ -332,8 +332,6 @@ def get_loss(self, logits, labels):
         sum_ = eps + tf.reduce_sum(flat_pred + flat_labels, axis=0)
         dice = 2 * intersection / sum_
 
-        dice = dice[1:]
-
         loss = 0
         for d in dice:
             loss += d