Updated thesis

src/trainers/diffusion_trainer.py

@@ -69,7 +69,7 @@ class DiffusionTrainer:
         self.model = model
         self.device = device
 
-        self.noise_steps = 30
+        self.noise_steps = 1000
         self.beta_start = 0.0001
         self.beta_end = 0.02
         self.ts_length = 96

src/training_scripts/diffusion_training.py

@@ -2,7 +2,7 @@ from src.utils.clearml import ClearMLHelper
 
 clearml_helper = ClearMLHelper(project_name="Thesis/NrvForecast")
 task = clearml_helper.get_task(
-    task_name="Diffusion Training: hidden_sizes=[256, 256], lr=0.0001, time_dim=8 + Load + PV + Wind + NP"
+    task_name="Diffusion Training: hidden_sizes=[1024, 1024, 1024], lr=0.0001, time_dim=8 + Load + PV + Wind + NP"
 )
 task.execute_remotely(queue_name="default", exit_process=True)
 
@@ -45,7 +45,7 @@ print("Input dim: ", inputDim)
 model_parameters = {
     "epochs": 15000,
     "learning_rate": 0.0001,
-    "hidden_sizes": [256, 256, 256],
+    "hidden_sizes": [1024, 1024, 1024],
     "time_dim": 8,
 }