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

View File

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