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

Autoregressive test score calculated on 96 values

Browse Source
This commit is contained in:
Victor Mylle
2023-11-13 23:14:07 +00:00
parent caeba6b841
commit 75f1f64c38
4 changed files with 45 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/trainers/trainer.py
View File

@@ -180,7 +180,7 @@ class Trainer:
transformed_train_loader, transformed_test_loader = self.data_processor.get_dataloaders(predict_sequence_length=self.model.output_size)
self.log_final_metrics(task, transformed_train_loader, train=True)
# self.log_final_metrics(task, transformed_train_loader, train=True)
self.log_final_metrics(task, transformed_test_loader, train=False)