Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'dev' of https://github.com/shahules786/enhancer into dev-hawk

This commit is contained in:
shahules786 2022-10-22 12:00:46 +05:30
parent 4b34cf6980 02192e5567
commit f492e44e6b
2 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
enhancer/data/dataset.py
View File

@ -77,7 +77,6 @@ class TaskDataset(pl.LightningDataModule):
if num_workers is None: if num_workers is None:
num_workers = multiprocessing.cpu_count() // 2 num_workers = multiprocessing.cpu_count() // 2
self.num_workers = num_workers self.num_workers = num_workers
print("num_workers-main", self.num_workers)
if valid_minutes > 0.0: if valid_minutes > 0.0:
self.valid_minutes = valid_minutes self.valid_minutes = valid_minutes
else: else:
@ -102,9 +101,6 @@ class TaskDataset(pl.LightningDataModule):
self.train_data = self.prepare_traindata(train_data) self.train_data = self.prepare_traindata(train_data)
self._validation = self.prepare_mapstype(self.val_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_clean = os.path.join(self.root_dir, self.files.test_clean)
test_noisy = os.path.join(self.root_dir, self.files.test_noisy) test_noisy = os.path.join(self.root_dir, self.files.test_noisy)
@ -349,7 +345,6 @@ class EnhancerDataset(TaskDataset):
return { return {
"clean": clean_segment, "clean": clean_segment,
"noisy": noisy_segment, "noisy": noisy_segment,
"name": file_dict["clean"].split("/")[-1] + "->" + str(start_time),
} }
def train__len__(self): def train__len__(self):

4
enhancer/loss.py
View File

@ -131,8 +131,8 @@ class Pesq:
pesq_values.append( pesq_values.append(
pesq( pesq(
self.sr, self.sr,
target_.squeeze().detach().numpy(), target_.squeeze().detach().cpu().numpy(),
pred.squeeze().detach().numpy(), pred.squeeze().detach().cpu().numpy(),
self.mode, self.mode,
) )
) )