switched to mean on regularization

Jhsmit · Jul 17, 2020 · 90a1b0c · 90a1b0c
1 parent 2ad393c
commit 90a1b0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhdx/fitting_tf.py b/pyhdx/fitting_tf.py
@@ -22,9 +22,9 @@ def __call__(self, x):
         regularization = 0.
         diff = x[:-1] - x[1:]
         if self.l1:
-            regularization += self.l1*math_ops.reduce_sum(math_ops.abs(diff))
+            regularization += self.l1*tf.math.reduce_mean(math_ops.abs(diff))
         if self.l2:
-            regularization += self.l2*math_ops.reduce_sum(math_ops.square(diff))  # should we take the sqrt of the sum?
+            regularization += self.l2*tf.math.reduce_mean(math_ops.square(diff))  # should we take the sqrt of the sum?
 
         return regularization