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