From 4ed7fe3ce55b9206f96973ec1c4530685cb73589 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Mon, 17 Oct 2022 11:03:09 +0530 Subject: [PATCH] dns 30 hrs demucs --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/DNS-2020.yaml | 2 +- enhancer/cli/train_config/trainer/default.yaml | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index 8d0ab14..9c464ac 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,6 +1,6 @@ defaults: - model : Demucs - - dataset : Vctk + - dataset : DNS-2020 - optimizer : Adam - hyperparameters : default - trainer : default diff --git a/enhancer/cli/train_config/dataset/DNS-2020.yaml b/enhancer/cli/train_config/dataset/DNS-2020.yaml index 4c6bfee..f23d412 100644 --- a/enhancer/cli/train_config/dataset/DNS-2020.yaml +++ b/enhancer/cli/train_config/dataset/DNS-2020.yaml @@ -1,5 +1,5 @@ _target_: enhancer.data.dataset.EnhancerDataset -root_dir : /scratch/c.sistc3/MS-SNSD/DNS10 +root_dir : /scratch/c.sistc3/MS-SNSD/DNS30 name : dns-2020 duration : 5.0 sampling_rate: 16000 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 36a7912..d4917e8 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -23,7 +23,7 @@ limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 log_every_n_steps: 100 -max_epochs: 300 +max_epochs: 250 max_time: 00:47:00:00 min_epochs: 1 min_steps: null