test loss function

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