diff --git a/climax/core/loss.py b/climax/core/loss.py index 236c2c0e22ea9986bedd901186edc0a86ff502b8..c2cc811cb3d2ffaf3f8c81bbc46a1be6ac8be949 100644 --- a/climax/core/loss.py +++ b/climax/core/loss.py @@ -139,8 +139,7 @@ class BernoulliWeibullLoss(BernoulliLoss): # clip shape to (0, 10) # NOTE: in general shape in (0, +infinity), clipping is required for # numerical stability - shape = torch.clamp( - torch.exp(y_pred[:, 1, ...].squeeze()[mask][~mask_p]), max=10) + shape = torch.exp(y_pred[:, 1, ...].squeeze()[mask][~mask_p]) # negative log-likelihood function of Bernoulli-Weibull distribution loss = torch.zeros_like(y_true)