test loss function

Arno1235 · Mar 20, 2023 · fc3d3ce · fc3d3ce
1 parent 6ec1e3d
commit fc3d3ce
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/unet_2d.py b/unet_2d.py
@@ -307,6 +307,9 @@ def get_loss(self, logits, labels):
 
         loss_map = tf.nn.softmax_cross_entropy_with_logits(logits=flat_logits, labels=flat_labels)
         print(f'TEST loss map: {loss_map}, size: {len(loss_map)}')
+        probs = tf.nn.softmax(logits, axis=-1)
+        flat_probs = tf.reshape(probs, [-1, self.n_class])
+        print(f'TEST probs: {probs}, flat_probs: {flat_probs}')
 
         if self.weight_type is None:
             loss = tf.reduce_mean(loss_map)