diff --git a/cli/train_config/config.yaml b/cli/train_config/config.yaml
index 6b5d98e..61551bd 100644
--- a/cli/train_config/config.yaml
+++ b/cli/train_config/config.yaml
@@ -1,5 +1,5 @@
 defaults:
-  - model : Demucs
+  - model : WaveUnet
   - dataset : Vctk
   - optimizer : Adam
   - hyperparameters : default
diff --git a/cli/train_config/trainer/default.yaml b/cli/train_config/trainer/default.yaml
index 560305b..ab4e273 100644
--- a/cli/train_config/trainer/default.yaml
+++ b/cli/train_config/trainer/default.yaml
@@ -2,7 +2,7 @@ _target_: pytorch_lightning.Trainer
 accelerator: auto
 accumulate_grad_batches: 1
 amp_backend: native
-auto_lr_find: False
+auto_lr_find: True
 auto_scale_batch_size: False
 auto_select_gpus: True
 benchmark: False