From d318480b7c4fdda9b4ea4b8cfe7a9df58fbccd47 Mon Sep 17 00:00:00 2001 From: "Michael Pilosov, PhD" Date: Sun, 28 Jan 2024 02:58:30 +0000 Subject: [PATCH] circle norm bad for supervise? --- model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model.py b/model.py index 477b094..dd6029d 100644 --- a/model.py +++ b/model.py @@ -58,8 +58,8 @@ class ColorTransformerModel(L.LightningModule): alpha = self.hparams.alpha # N = len(outputs) - distance = circle_norm(outputs, labels).mean() - # distance = torch.norm(outputs - labels).mean() + # distance = circle_norm(outputs, labels).mean() + distance = torch.norm(outputs - labels).mean() # Backprop with this: loss = (1 - alpha) * p_loss + alpha * distance