Sped up sampling 20x

2023-11-25 18:09:42 +00:00
parent 5de3f64a1a
commit 300f268286
10 changed files with 498 additions and 238 deletions
--- a/src/losses/crps_metric.py
+++ b/src/losses/crps_metric.py
@@ -15,7 +15,7 @@ class CRPSLoss(nn.Module):
        # preds shape: [batch_size, num_quantiles]

        # unsqueeze target
-        target = target.unsqueeze(-1)
+        # target = target.unsqueeze(-1)

        mask = (preds > target).float()
        test = self.quantiles_tensor - mask