Updated some stuff

2024-03-20 22:16:19 +01:00
parent dad64d00be
commit e780b46af7
6 changed files with 22 additions and 13 deletions
--- a/src/training_scripts/autoregressive_quantiles.py
+++ b/src/training_scripts/autoregressive_quantiles.py
@@ -114,17 +114,6 @@ trainer = AutoRegressiveQuantileTrainer(
    debug=False,
 )

-# trainer = NonAutoRegressiveQuantileRegression(
-#     model,
-#     inputDim,
-#     optimizer,
-#     data_processor,
-#     quantiles,
-#     "cuda",
-#     policy_evaluator=policy_evaluator,
-#     debug=False,
-# )
-
 trainer.add_metrics_to_track(
    [PinballLoss(quantiles), MSELoss(), L1Loss(), CRPSLoss(quantiles)]
 )
--- a/src/training_scripts/diffusion_training.py
+++ b/src/training_scripts/diffusion_training.py
@@ -18,6 +18,7 @@ from src.policies.PolicyEvaluator import PolicyEvaluator
 #### Data Processor ####
 data_config = DataConfig()
 data_config.NRV_HISTORY = True
+
 data_config.LOAD_HISTORY = False
 data_config.LOAD_FORECAST = False