From e1602567d6189e13e1856e3e4050036f094cbbdf Mon Sep 17 00:00:00 2001
From: Hunor Laczko <luni741@gmail.com>
Date: Thu, 16 Jun 2022 15:28:23 +0000
Subject: [PATCH] use mean instead of sum

---
 losses.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/losses.py b/losses.py
index d9245eb..8329b0a 100644
--- a/losses.py
+++ b/losses.py
@@ -124,7 +124,7 @@ def collision_loss(V, B, mask, thr=.004):
     D = V - tf.gather_nd(B, idx)
     b_vn = tf.gather_nd(B_vn, idx)
     dot = tf.einsum('abc,abc->ab', D, b_vn)
-    loss = tf.reduce_sum((tf.minimum(dot - thr, 0)  * mask) ** 2)
+    loss = tf.reduce_mean((tf.minimum(dot - thr, 0)  * mask) ** 2)
 
     return loss
 
@@ -147,7 +147,7 @@ def edge_loss(x, y, mask):
     y_e = tf.sqrt(y_e + 1e-9)
     d_e = y_e - x_e
     d_e = d_e * e_mask
-    loss = tf.reduce_sum(d_e ** 2)
+    loss = tf.reduce_mean(d_e ** 2)
     return loss