Changes to allow remote execution

2023-11-25 23:36:57 +00:00
parent 300f268286
commit a8db70e86d
7 changed files with 4500091 additions and 4 deletions
--- a/src/losses/pinball_loss.py
+++ b/src/losses/pinball_loss.py
@@ -6,7 +6,6 @@ class PinballLoss(nn.Module):
    def __init__(self, quantiles):
        super(PinballLoss, self).__init__()
        self.quantiles_tensor = torch.tensor(quantiles, dtype=torch.float32)
-        self.quantiles = self.quantiles_tensor.tolist()

    def forward(self, pred, target):
        error = target - pred