From 20c12556fff9e577045edecd700db1ef33ffe2a2 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 21 Oct 2022 16:25:24 +0530 Subject: [PATCH] debug --- enhancer/data/dataset.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index cc89083..fb4d04c 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -100,6 +100,9 @@ class TaskDataset(pl.LightningDataModule): self.train_data = self.prepare_traindata(train_data) self._validation = self.prepare_mapstype(self.val_data) + print( + "train_data_size", sum([len(item) for item in self.train_data]) + ) test_clean = os.path.join(self.root_dir, self.files.test_clean) test_noisy = os.path.join(self.root_dir, self.files.test_noisy)