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/trainers/quantile_trainer.py
+++ b/src/trainers/quantile_trainer.py
@@ -67,9 +67,10 @@ class AutoRegressiveQuantileTrainer(AutoRegressiveTrainer):
        clearml_helper: ClearMLHelper = None,
        debug: bool = True,
    ):
+
+        self.quantiles = quantiles
        quantiles_tensor = torch.tensor(quantiles)
        quantiles_tensor = quantiles_tensor.to(device)
-        self.quantiles = quantiles

        criterion = PinballLoss(quantiles=quantiles_tensor)
        super().__init__(