Updated thesis

2024-05-03 18:13:01 +02:00
parent 2597577d3d
commit 074e654b8a
34 changed files with 2194 additions and 596 deletions
--- a/src/trainers/diffusion_trainer.py
+++ b/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