Handle infinite values for Weibull.

climax/core/loss.py

@@ -199,4 +199,5 @@ class BernoulliWeibullLoss(NaNLoss):
                           (gshape - 1) * torch.log(y_true + self.epsilon) -
                           (y_true / (gscale + self.epsilon)) ** gshape)
 
-        return self.reduce(loss)
+        # handle +-infinity values resulting from initialization
+        return self.reduce(loss[torch.isfinite(loss)])