From 3df4b2713277316bbd54f3f975d6fc30f564def9 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Tue, 18 Oct 2022 21:29:55 +0530 Subject: [PATCH] dns --- enhancer/cli/train_config/dataset/DNS-2020.yaml | 5 +++-- enhancer/cli/train_config/model/Demucs.yaml | 4 ++-- 2 files changed, 5 insertions(+), 4 deletions(-) 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