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

debug

This commit is contained in:
shahules786 2022-10-21 16:25:24 +05:30
parent 9c7a650130
commit 20c12556ff
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
enhancer/data/dataset.py
View File

@ -100,6 +100,9 @@ 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)