Added new training scripts

2023-11-27 14:55:22 +00:00
parent 5e87165dbb
commit c1152ff96c
7 changed files with 37 additions and 36 deletions
--- a/src/losses/crps_metric.py
+++ b/src/losses/crps_metric.py
@@ -18,6 +18,7 @@ class CRPSLoss(nn.Module):
        # target = target.unsqueeze(-1)

        mask = (preds > target).float()
+        self.quantiles_tensor = self.quantiles_tensor.to(preds.device)
        test = self.quantiles_tensor - mask
        # square them
        test = test * test