update dummy loss to take mean not sum

dummy run wasn't doing anything reasonable
janelia-cellmap · Feb 18, 2025 · 02e5811 · 02e5811
1 parent d523f36
commit 02e5811
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dacapo/experiments/tasks/losses/dummy_loss.py b/dacapo/experiments/tasks/losses/dummy_loss.py
@@ -44,4 +44,4 @@ def compute(self, prediction, target, weight=None):
             It is used to test the training loop and the loss calculation.
         """
 
-        return abs(prediction - target).sum()
+        return abs(prediction - target).mean()