From 7b0e7e2312be6faf603969bfe6c5441ab89c05e6 Mon Sep 17 00:00:00 2001 From: shahules786 Date: Fri, 30 Sep 2022 10:15:28 +0530 Subject: [PATCH] change config --- cli/train_config/config.yaml | 2 +- cli/train_config/hyperparameters/default.yaml | 2 +- cli/train_config/trainer/default.yaml | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/cli/train_config/config.yaml b/cli/train_config/config.yaml index 61551bd..6b5d98e 100644 --- a/cli/train_config/config.yaml +++ b/cli/train_config/config.yaml @@ -1,5 +1,5 @@ defaults: - - model : WaveUnet + - model : Demucs - dataset : Vctk - optimizer : Adam - hyperparameters : default diff --git a/cli/train_config/hyperparameters/default.yaml b/cli/train_config/hyperparameters/default.yaml index 4931c7c..04b099b 100644 --- a/cli/train_config/hyperparameters/default.yaml +++ b/cli/train_config/hyperparameters/default.yaml @@ -1,4 +1,4 @@ loss : mse metric : mae -lr : 0.001 +lr : 0.0001 num_epochs : 100 diff --git a/cli/train_config/trainer/default.yaml b/cli/train_config/trainer/default.yaml index 633c6ba..560305b 100644 --- a/cli/train_config/trainer/default.yaml +++ b/cli/train_config/trainer/default.yaml @@ -9,7 +9,7 @@ benchmark: False check_val_every_n_epoch: 1 detect_anomaly: False deterministic: False -devices: auto +devices: -1 enable_checkpointing: True enable_model_summary: True enable_progress_bar: True @@ -22,7 +22,7 @@ limit_predict_batches: 1.0 limit_test_batches: 1.0 limit_train_batches: 1.0 limit_val_batches: 1.0 -log_every_n_steps: 50 +log_every_n_steps: 10 max_epochs: 100 max_steps: null max_time: null