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

w/o striding

This commit is contained in:
shahules786 2022-10-22 18:09:55 +05:30
parent 4cfe7a2463
commit 96c934e96f
3 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
enhancer/cli/train_config/config.yaml
View File

@ -1,6 +1,6 @@
defaults: defaults:
- model : Demucs - model : Demucs
- dataset : DNS-2020 - dataset : Vctk
- optimizer : Adam - optimizer : Adam
- hyperparameters : default - hyperparameters : default
- trainer : default - trainer : default

1
enhancer/cli/train_config/dataset/Vctk.yaml
View File

@ -2,7 +2,6 @@ _target_: enhancer.data.dataset.EnhancerDataset
name : vctk name : vctk
root_dir : /scratch/c.sistc3/DS_10283_2791 root_dir : /scratch/c.sistc3/DS_10283_2791
duration : 4.5 duration : 4.5
stride : 0.5
sampling_rate: 16000 sampling_rate: 16000
batch_size: 128 batch_size: 128
valid_minutes : 10 valid_minutes : 10

4
enhancer/data/dataset.py
View File

@ -251,7 +251,7 @@ class EnhancerDataset(TaskDataset):
files: Files, files: Files,
valid_minutes=5.0, valid_minutes=5.0,
duration=1.0, duration=1.0,
stride=0.5, stride=None,
sampling_rate=48000, sampling_rate=48000,
matching_function=None, matching_function=None,
batch_size=32, batch_size=32,
@ -317,7 +317,7 @@ class EnhancerDataset(TaskDataset):
return self.prepare_segment(*self._test[idx]) return self.prepare_segment(*self._test[idx])
def prepare_segment(self, file_dict: dict, start_time: float): def prepare_segment(self, file_dict: dict, start_time: float):
print(file_dict["clean"].split("/")[-1], "->", start_time)
clean_segment = self.audio( clean_segment = self.audio(
file_dict["clean"], offset=start_time, duration=self.duration file_dict["clean"], offset=start_time, duration=self.duration
) )