back to "correct" metric

10 months ago · 5687f30818
1 changed files with 1 additions and 1 deletions
--- a/losses.py
+++ b/losses.py
@ -40,7 +40,7 @@ def preservation_loss(inputs, outputs, target_inputs=None, target_outputs=None):
    # print(rgb_norm)
    # Calculate 1D Space Norm (modulo 1 to account for circularity)
-    transformed_norm = circle_norm(outputs, target_outputs)  # * 2
+    transformed_norm = circle_norm(outputs, target_outputs) * 2
    diff = torch.pow(rgb_norm - transformed_norm, 2)
    N = len(outputs)