From 6eb905c1bb67590e62ff2dc1598ee3a1257aa06c Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 12:00:18 +0530 Subject: [PATCH 1/2] 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): From 02192e556729f38ecfff00c1415bb9d251b3b96b Mon Sep 17 00:00:00 2001 From: shahules786 Date: Sat, 22 Oct 2022 12:00:30 +0530 Subject: [PATCH 2/2] to cpu --- enhancer/loss.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/enhancer/loss.py b/enhancer/loss.py index f2be8df..2150699 100644 --- a/enhancer/loss.py +++ b/enhancer/loss.py @@ -131,8 +131,8 @@ class Pesq: pesq_values.append( pesq( self.sr, - target_.squeeze().detach().numpy(), - pred.squeeze().detach().numpy(), + target_.squeeze().detach().cpu().numpy(), + pred.squeeze().detach().cpu().numpy(), self.mode, ) )