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

Fixed crps + more inputs

Browse Source
This commit is contained in:
Victor Mylle
2023-12-05 00:08:17 +00:00
parent 120b6aa5bd
commit d3bf04d68c
13 changed files with 128426 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/trainers/trainer.py
View File

@@ -166,11 +166,17 @@ class Trainer:
if hasattr(self, "plot_quantile_percentages"):
self.plot_quantile_percentages(
task, train_loader, True, epoch
task, train_loader, True, epoch, False
)
# self.plot_quantile_percentages(
# task, train_loader, True, epoch, True
# )
self.plot_quantile_percentages(
task, test_loader, False, epoch
task, test_loader, False, epoch, False
)
# self.plot_quantile_percentages(
# task, test_loader, False, epoch, True
# )
if task:
self.finish_training(task=task)