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

rmv print statements

This commit is contained in:
shahules786 2022-10-22 12:00:18 +05:30
parent 9f658424a6
commit 6eb905c1bb
1 changed files with 0 additions and 5 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):