Skip to content
Snippets Groups Projects
Commit 97a7518f authored by Tamino Huxohl's avatar Tamino Huxohl
Browse files

fix l1 loss usage

parent 08480c2a
No related branches found
No related tags found
No related merge requests found
......@@ -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()
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment