From d10f8a5ff6631b01b734467afe6a8fd5649e3286 Mon Sep 17 00:00:00 2001 From: Victor Mylle Date: Mon, 12 Feb 2024 09:54:56 +0000 Subject: [PATCH] Clamping diffusion output --- src/training_scripts/diffusion_training.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/training_scripts/diffusion_training.py b/src/training_scripts/diffusion_training.py index f68e786..9a1c2a3 100644 --- a/src/training_scripts/diffusion_training.py +++ b/src/training_scripts/diffusion_training.py @@ -54,8 +54,8 @@ model_parameters = { 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"]) -# model = GRUDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[2], time_dim=model_parameters["time_dim"], gru_hidden_size=256) +# model = SimpleDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[1], time_dim=model_parameters["time_dim"]) +model = GRUDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[2], time_dim=model_parameters["time_dim"], gru_hidden_size=256) print("Starting training ...")