Changes to allow remote execution

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__(

src/trainers/trainer.py

@@ -76,6 +76,7 @@ class Trainer:
         task.connect(self.optimizer, name="optimizer")
         task.connect(self.criterion, name="criterion")
         task.connect(self.data_processor, name="data_processor")
+        task.connect(self, name="trainer")
 
         task.connect(self.data_processor.data_config, name="data_features")