Fix bug on loss functions

ClementPinard · Apr 9, 2019 · 63af665 · 63af665
1 parent 389dbe1
commit 63af665
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/loss_functions.py b/loss_functions.py
@@ -54,7 +54,7 @@ def one_scale(depth, explainability_mask):
         total_loss += loss
         warped_results.append(warped)
         diff_results.append(diff)
-    return loss, warped_results, diff_results
+    return total_loss, warped_results, diff_results
 
 
 def explainability_loss(mask):