From 5687f308183bd7e724a6e39d1ef914bb17b4e9b9 Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Sun, 28 Jan 2024 08:32:05 +0000 Subject: [PATCH] back to "correct" metric --- losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/losses.py b/losses.py index b5fa227..e68e465 100644 --- 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)