Fix simple_loss

zincware · Apr 10, 2024 · 99eff3e · 99eff3e
1 parent 7ba5de5
commit 99eff3e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/znnl/loss_functions/simple_loss.py b/znnl/loss_functions/simple_loss.py
@@ -73,6 +73,6 @@ def __call__(
                 total loss of all points based on the similarity measurement
         """
         if mask is not None:
-            return np.sum(self.metric(point_1, point_2) * mask, axis=0)
+            return np.mean(self.metric(point_1, point_2) * mask, axis=0)
         else:
-            return np.sum(self.metric(point_1, point_2), axis=0)
+            return np.mean(self.metric(point_1, point_2), axis=0)