Fixing some stuff

2023-12-30 15:22:32 +00:00
parent ef8b5f49ac
commit c26ae76951
6 changed files with 107 additions and 33 deletions
--- a/src/training_scripts/autoregressive_quantiles.py
+++ b/src/training_scripts/autoregressive_quantiles.py
@@ -43,7 +43,7 @@ data_processor.set_full_day_skip(False)
 #### Hyperparameters ####
 data_processor.set_output_size(1)
 inputDim = data_processor.get_input_size()
-epochs = 400
+epochs = 300

 # add parameters to clearml
 quantiles = task.get_parameter("general/quantiles", cast=True)
@@ -58,7 +58,7 @@ else:

 model_parameters = {
    "learning_rate": 0.0001,
-    "hidden_size": 1024,
+    "hidden_size": 512,
    "num_layers": 2,
    "dropout": 0.2,
    "time_feature_embedding": 4,