From 58c42b1eae2db47012e06b722f1c7db294c0e5a1 Mon Sep 17 00:00:00 2001 From: Schobbejak Date: Mon, 10 Jun 2024 14:45:36 +0200 Subject: [PATCH] Rename dataset back to validation dataset --- epochalyst/pipeline/model/training/torch_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/epochalyst/pipeline/model/training/torch_trainer.py b/epochalyst/pipeline/model/training/torch_trainer.py index e7f1c63..95da046 100644 --- a/epochalyst/pipeline/model/training/torch_trainer.py +++ b/epochalyst/pipeline/model/training/torch_trainer.py @@ -826,7 +826,7 @@ def get_model_checkpoint_path(self, epoch: int) -> Path: return Path(f"{self.trained_models_directory}/{self.get_hash()}_checkpoint_{epoch}.pt") -class TrainTestDataset(Dataset[T_co]): +class TrainValidationDataset(Dataset[T_co]): """Dataset as a concatenation of multiple datasets. This class is useful to assemble different existing datasets.