From 8ff5cb13bdf08e2e45daa4bb12bdedc08f248a3f Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Tue, 24 Aug 2021 12:10:57 +0200
Subject: [PATCH] Fixed error in loss function: log(tau(alpha)).
---
climax/core/loss.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/climax/core/loss.py b/climax/core/loss.py
index 02e7a5b..5ed3af2 100644
--- a/climax/core/loss.py
+++ b/climax/core/loss.py
@@ -92,6 +92,7 @@ class BernoulliGammaLoss(NaNLoss):
torch.log(y_true + self.epsilon) -
y_true / (gscale + self.epsilon) -
gshape * torch.log(gscale + self.epsilon) -
- torch.lgamma(gshape + self.epsilon))
+ torch.log(torch.lgamma(gshape + self.epsilon)
+ + self.epsilon))
return - self.reduce(loss)
--
GitLab