From aaf0ae0e5fe1d403028ea7ea06f209fb9cfff0cf Mon Sep 17 00:00:00 2001
From: Victor Mylle <victor.mylle@@hotmail.com>
Date: Mon, 15 Jan 2024 11:49:30 +0000
Subject: [PATCH] Adding print statement

---
 src/training_scripts/diffusion_training.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/training_scripts/diffusion_training.py b/src/training_scripts/diffusion_training.py
index fd384f3..dfac0aa 100644
--- a/src/training_scripts/diffusion_training.py
+++ b/src/training_scripts/diffusion_training.py
@@ -55,6 +55,8 @@ model_parameters = task.connect(model_parameters, name="model_parameters")
 #### Model ####
 model = SimpleDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[1], time_dim=model_parameters["time_dim"])
 
+print("Starting training ...")
+
 #### Trainer ####
 trainer = DiffusionTrainer(model, data_processor, "cuda")
 trainer.train(model_parameters["epochs"], model_parameters["learning_rate"], task)
\ No newline at end of file