From 0e58691a2c683175e22b8be91479dcc772941f67 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 14 Oct 2022 12:45:34 +0530 Subject: [PATCH] demucs 250 --- enhancer/cli/train_config/config.yaml | 2 +- enhancer/cli/train_config/dataset/Vctk.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 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 97e2f0c..a4707bf 100644 --- a/enhancer/cli/train_config/dataset/Vctk.yaml +++ b/enhancer/cli/train_config/dataset/Vctk.yaml @@ -1,7 +1,7 @@ _target_: enhancer.data.dataset.EnhancerDataset name : vctk root_dir : /scratch/c.sistc3/DS_10283_2791 -duration : 1.0 +duration : 1.5 sampling_rate: 16000 batch_size: 256 valid_size : 0.05 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index beb497c..f3d0804 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: 500 +max_epochs: 250 max_time: null min_epochs: 1 min_steps: null