From 8d8eaa80d5d305841ac2bfbe5bba79f2018ed38f Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 28 Oct 2022 13:09:29 +0530 Subject: [PATCH] vctk+demucs --- enhancer/cli/train_config/hyperparameters/default.yaml | 6 +++--- enhancer/cli/train_config/trainer/default.yaml | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/enhancer/cli/train_config/hyperparameters/default.yaml b/enhancer/cli/train_config/hyperparameters/default.yaml index 7a9d9a3..09d3ebb 100644 --- a/enhancer/cli/train_config/hyperparameters/default.yaml +++ b/enhancer/cli/train_config/hyperparameters/default.yaml @@ -1,8 +1,8 @@ loss : mae metric : [stoi,pesq,si-sdr] lr : 0.0003 -ReduceLr_patience : 8 +ReduceLr_patience : 5 Early_stop : True -ReduceLr_factor : 0.5 -min_lr : 0.000001 +ReduceLr_factor : 0.25 +min_lr : 0.0000001 EarlyStopping_patience : 15 diff --git a/enhancer/cli/train_config/trainer/default.yaml b/enhancer/cli/train_config/trainer/default.yaml index 0217553..bd11ad7 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: 2 +devices: 1 enable_checkpointing: True enable_model_summary: True enable_progress_bar: True @@ -38,7 +38,7 @@ precision: 32 profiler: null reload_dataloaders_every_n_epochs: 0 replace_sampler_ddp: True -strategy: ddp +strategy: null sync_batchnorm: False tpu_cores: null track_grad_norm: -1