From f2561d7cf7f6cd1069310ddbdc4e0074da1e9b0d Mon Sep 17 00:00:00 2001 From: shahules786 Date: Thu, 20 Oct 2022 09:53:27 +0530 Subject: [PATCH] config --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/Vctk.yaml | 5 +++-- enhancer/cli/train_config/trainer/default.yaml | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/enhancer/cli/train_config/config.yaml b/enhancer/cli/train_config/config.yaml index c0b2cf6..8d0ab14 100644 --- a/enhancer/cli/train_config/config.yaml +++ b/enhancer/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : WaveUnet + - model : Demucs - dataset : Vctk - optimizer : Adam - hyperparameters : default diff --git a/enhancer/cli/train_config/dataset/Vctk.yaml b/enhancer/cli/train_config/dataset/Vctk.yaml index df50da2..c128404 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -1,9 +1,10 @@ _target_: enhancer.data.dataset.EnhancerDataset name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 -duration : 1.0 +duration : 4.5 +stride : 0.5 sampling_rate: 16000 -batch_size: 128 +batch_size: 16 files: train_clean : clean_trainset_28spk_wav diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index dfc020f..01914e4 100644 --- a/enhancer/cli/train_config/trainer/default.yaml +++ b/enhancer/cli/train_config/trainer/default.yaml @@ -9,7 +9,7 @@ benchmark: False check_val_every_n_epoch: 1 detect_anomaly: False deterministic: False -devices: -1 +devices: 1 enable_checkpointing: True enable_model_summary: True enable_progress_bar: True