VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding print statement

Browse Source
This commit is contained in:
Victor Mylle
2024-01-15 11:49:30 +00:00
parent a977021dfc
commit aaf0ae0e5f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/training_scripts/diffusion_training.py
View File

@@ -55,6 +55,8 @@ model_parameters = task.connect(model_parameters, name="model_parameters")
#### Model #### #### Model ####
model = SimpleDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[1], time_dim=model_parameters["time_dim"]) model = SimpleDiffusionModel(96, model_parameters["hidden_sizes"], other_inputs_dim=inputDim[1], time_dim=model_parameters["time_dim"])
print("Starting training ...")
#### Trainer #### #### Trainer ####
trainer = DiffusionTrainer(model, data_processor, "cuda") trainer = DiffusionTrainer(model, data_processor, "cuda")
trainer.train(model_parameters["epochs"], model_parameters["learning_rate"], task) trainer.train(model_parameters["epochs"], model_parameters["learning_rate"], task)