Add epsilon for numerical stability.

climax/core/loss.py

@@ -82,7 +82,7 @@ class BernoulliGammaLoss(NaNLoss):
         # negative log-likelihood function of Bernoulli-Gamma distribution
 
         # Bernoulli contribution
-        loss = (1 - p_true) * torch.log(1 - p_pred)
+        loss = (1 - p_true) * torch.log(1 - p_pred + self.epsilon)
 
         # Gamma contribution
         loss += p_true * (torch.log(p_pred + self.epsilon) + (gshape - 1) *