diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index d498be5..b28e3ef 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -1,9 +1,10 @@ _target_: enhancer.data.dataset.EnhancerDataset -root_dir : /scratch/c.sistc3/MS-SNSD/DNS30 +root_dir : /scratch/c.sistc3/MS-SNSD/DNS10 name : dns-2020 duration : 5.0 sampling_rate: 16000 -batch_size: 64 +batch_size: 128 +stride: 2 valid_minutes: 30.0 files: train_clean : CleanSpeech_training diff --git a/enhancer/cli/train_config/model/Demucs.yaml b/enhancer/cli/train_config/model/Demucs.yaml index aec1592..b27abbd 100644 --- a/enhancer/cli/train_config/model/Demucs.yaml +++ b/enhancer/cli/train_config/model/Demucs.yaml @@ -1,11 +1,11 @@ _target_: enhancer.models.demucs.Demucs num_channels: 1 -resample: 2 +resample: 4 sampling_rate : 16000 encoder_decoder: depth: 5 - initial_output_channels: 64 + initial_output_channels: 48 kernel_size: 8 stride: 4 growth_factor: 2