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