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