From 97a7518f5976a9ba9c3760a823c6240f5da485ce Mon Sep 17 00:00:00 2001
From: Tamino Huxohl <thuxohl@techfak.uni-bielefeld.de>
Date: Fri, 16 Dec 2022 16:13:10 +0100
Subject: [PATCH] fix l1 loss usage
---
mu_map/training/distance.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mu_map/training/distance.py b/mu_map/training/distance.py
index 5c44605..d40ea99 100644
--- a/mu_map/training/distance.py
+++ b/mu_map/training/distance.py
@@ -36,7 +36,7 @@ class Training(AbstractTraining):
def _eval_batch(self, recons: torch.Tensor, mu_maps: torch.Tensor) -> float:
mu_maps_predicted = self.model(recons)
- loss = torch.nn.functional.loss.l1(mu_maps_predicted, mu_maps)
+ loss = torch.nn.functional.l1_loss(mu_maps_predicted, mu_maps)
return loss.item()
--
GitLab