adjustment fot CCS, notebook folder

This commit is contained in:
Steffen Illium
2021-03-22 16:43:19 +01:00
parent 78b3139d1a
commit c12f3866c8
6 changed files with 156 additions and 29 deletions

View File

@ -116,6 +116,7 @@ class CCSLibrosaDatamodule(_BaseDataModule):
for sub_dataset in results.get():
dataset.append(sub_dataset[0])
datasets[data_option] = ConcatDataset(dataset)
print(f'{data_option}-dataset prepared.')
self.datasets = datasets
return datasets
@ -133,6 +134,7 @@ class CCSLibrosaDatamodule(_BaseDataModule):
for row in all_rows:
mel_dataset, class_id, _ = self._build_subdataset(row)
dataset.append(mel_dataset)
print(f'{data_option}-dataset prepared!')
datasets[data_option] = ConcatDataset(dataset)
# Build Weighted Sampler for train and val
@ -158,6 +160,7 @@ class CCSLibrosaDatamodule(_BaseDataModule):
samplers[data_option] = None
self.datasets = datasets
self.samplers = samplers
print(f'Dataset {self.__class__.__name__} setup done.')
return datasets
def purge(self):