adjustment fot CCS, notebook folder
This commit is contained in:
@ -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):
|
||||
|
Reference in New Issue
Block a user