Updated training scripts

2024-03-18 12:15:06 +01:00
parent 34335cd9fe
commit 1a8e735cbc
10 changed files with 487 additions and 308 deletions
--- a/src/trainers/trainer.py
+++ b/src/trainers/trainer.py
@@ -261,8 +261,7 @@ class Trainer:
        self.model.eval()

        # set full day skip
-        self.data_processor.set_full_day_skip(True)
-        train_loader, test_loader = self.data_processor.get_dataloaders(
+        _, test_loader = self.data_processor.get_dataloaders(
            predict_sequence_length=self.model.output_size
        )