diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 9c464ac..8d0ab14 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,6 +1,6 @@ defaults: - model : Demucs - - dataset : DNS-2020 + - dataset : Vctk - optimizer : Adam - hyperparameters : default - trainer : default diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index 849e192..0b5ba7f 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -2,7 +2,6 @@ _target_: enhancer.data.dataset.EnhancerDataset name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 duration : 4.5 -stride : 0.5 sampling_rate: 16000 batch_size: 128 valid_minutes : 10 diff --git a/enhancer/data/dataset.py b/enhancer/data/dataset.py index ddecc8a..624a796 100644 --- a/enhancer/data/dataset.py +++ b/enhancer/data/dataset.py @@ -251,7 +251,7 @@ class EnhancerDataset(TaskDataset): files: Files, valid_minutes=5.0, duration=1.0, - stride=0.5, + stride=None, sampling_rate=48000, matching_function=None, batch_size=32, @@ -317,7 +317,7 @@ class EnhancerDataset(TaskDataset): return self.prepare_segment(*self._test[idx]) def prepare_segment(self, file_dict: dict, start_time: float): - print(file_dict["clean"].split("/")[-1], "->", start_time) + clean_segment = self.audio( file_dict["clean"], offset=start_time, duration=self.duration )