diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index b28e3ef..3d9177d 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -3,7 +3,7 @@ root_dir : /scratch/c.sistc3/MS-SNSD/DNS10 name : dns-2020 duration : 5.0 sampling_rate: 16000 -batch_size: 128 +batch_size: 64 stride: 2 valid_minutes: 30.0 files: diff --git a/enhancer/cli/train_config/model/Demucs.yaml b/enhancer/cli/train_config/model/Demucs.yaml index b27abbd..32afb4a 100644 --- a/enhancer/cli/train_config/model/Demucs.yaml +++ b/enhancer/cli/train_config/model/Demucs.yaml @@ -1,6 +1,6 @@ _target_: enhancer.models.demucs.Demucs num_channels: 1 -resample: 4 +resample: 2 sampling_rate : 16000 encoder_decoder: