test loss function

Arno1235 · Mar 22, 2023 · 32dc51b · 32dc51b
1 parent 744c111
commit 32dc51b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/unet_2d.py b/unet_2d.py
@@ -319,9 +319,9 @@ def get_loss(self, logits, labels):
 
             if self.weight_type == 'feedback':
 
-                flat_labels = tf.Variable(flat_labels)
+                # flat_labels = tf.Variable(flat_labels)
                 for i in range(len(flat_labels)):
-                    # print(flat_labels[i][0])
+                    print(flat_labels[i])
                     flat_labels[i][0].assign(0)
 
                 weight_map = feedback_weight_map(flat_probs, flat_labels, 3, 100)