Non autregressive gru model load

src/trainers/quantile_trainer.py

@@ -633,6 +633,7 @@ class NonAutoRegressiveQuantileRegression(Trainer):
     def debug_plots(self, task, train: bool, data_loader, sample_indices, epoch):
         for actual_idx, idx in sample_indices.items():
             features, target, _ = data_loader.dataset[idx]
+            print(features.shape, target.shape)
 
             features = features.to(self.device)
             target = target.to(self.device)