test

Arno1235 · Mar 20, 2023 · 6180979 · 6180979
1 parent fd8de7e
commit 6180979
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/unet_2d.py b/unet_2d.py
@@ -315,7 +315,9 @@ def get_loss(self, logits, labels):
             else:
                 raise ValueError("Unknown weight type: "%self.weight_type)
 
-            loss = tf.reduce_mean(tf.multiply(loss_map, weight_map)) 
+            loss = tf.reduce_mean(tf.multiply(loss_map, weight_map))
+
+        print(f'TEST loss: {loss}') # TEST
 
         return loss