From 6eb905c1bb67590e62ff2dc1598ee3a1257aa06c Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 12:00:18 +0530 Subject: [PATCH] rmv print statements --- enhancer/data/dataset.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index fad1e92..7f7ae67 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -77,7 +77,6 @@ class TaskDataset(pl.LightningDataModule): if num_workers is None: num_workers = multiprocessing.cpu_count() // 2 self.num_workers = num_workers - print("num_workers-main", self.num_workers) if valid_minutes > 0.0: self.valid_minutes = valid_minutes else: @@ -102,9 +101,6 @@ 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) @@ -349,7 +345,6 @@ class EnhancerDataset(TaskDataset): return { "clean": clean_segment, "noisy": noisy_segment, - "name": file_dict["clean"].split("/")[-1] + "->" + str(start_time), } def train__len__(self):