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

Saving diffusion model on better CRPS score

Browse Source
This commit is contained in:
Victor Mylle
2023-12-29 10:48:07 +00:00
parent 81231b9266
commit da3ab3d5b3
2 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/training_scripts/diffusion_training.py
View File

@@ -35,7 +35,7 @@ data_config.DAY_OF_WEEK = False
data_config.NOMINAL_NET_POSITION = True
data_config = Task.connect(data_config, name="data_features")
data_config = task.connect(data_config, name="data_features")
data_processor = DataProcessor(data_config, path="", lstm=False)
data_processor.set_batch_size(8192)