test loss function

Arno1235 · Mar 22, 2023 · a27c0b5 · a27c0b5
1 parent c9d9234
commit a27c0b5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/unet_2d.py b/unet_2d.py
@@ -319,11 +319,10 @@ def get_loss(self, logits, labels):
 
             if self.weight_type == 'feedback':
 
-                # flat_labels = tf.Variable(flat_labels)
+                flat_labels = flat_labels.numpy()
                 for i in range(len(flat_labels)):
                     # print(flat_labels[i])
-                    flat_labels[i] = tf.Variable(flat_labels[i]).numpy()
-                    flat_labels[i].numpy()[0] = 0
+                    flat_labels[i][0] = 0
 
                 weight_map = feedback_weight_map(flat_probs, flat_labels, 3, 100)
                 print(f'wm {np.shape(weight_map)}')