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:
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):

4
enhancer/loss.py
View File

@ -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,
)
)